程序员进阶之算法练习(八)附两道搜狐笔试题

前言

前面讲了那么多算法的重要性。口说无凭,这次带上两道搜狐今年的笔试题。
这里先附上两道搜狐题目的大意:
题目一:
《宝石》
有一串宝石首尾相连,用一个大写字母表示一个宝石;
现在需要从这一串宝石中截取一段宝石,要求这一段宝石包含ABCDE这5种字母;求剩下最多有多少个宝石?
题目二:
《袋鼠》
有n个弹簧排成一列,袋鼠起始位置在第一个弹簧;
输入n个数字,代表n个弹簧的力量;
弹簧的力量为5表示可以往后跳最多5个弹簧;
问袋鼠到达第n个弹簧的最小弹跳次数?
答案看最后的附加题部分。

正文

这次正文有5道题,是CF707-DIV2的题目。

  • 为了不剧透,把对题目的总结放到最后面。

看完题目大意,先思考,再看解析;觉得题目大意不清晰,点击题目链接看原文。

文集:
程序员进阶之算法练习(一)
程序员进阶之算法练习(二)
程序员进阶之算法练习(三)
程序员进阶之算法练习(四)
程序员进阶之算法练习(五)
程序员进阶之算法练习(六)
程序员进阶之算法练习(七)
代码地址

A.Brain's Photos

题目链接
题目大意:输入n * m个字符,字符中存在C M Y为混色,否则为黑白,输出对应的描述。
'C' (cyan)
'M' (magenta)
'Y' (yellow)
'W' (white)
'G' (grey)
'B' (black)
n and m (1 ≤ n, m ≤ 100)

Examples
input
2 2
C M
Y Y
output
#Color

input
3 2
W W
W W
B B
output
\ #Black&White

代码实现

    int n, m, ret = 1;
    cin >> n >> m;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            char str[10];
            scanf("%s", str);
            if (str[0] == 'C' || str[0] == 'M' || str[0] == 'Y') {
                ret = 0;
            }
        }
    }
    if (!ret) {
        cout << "#Color";
    }
    else {
        cout << "#Black&White";
    }

题目解析
难度在读题,黑白不仅仅是W B,还有G。

B. Bakery

题目链接
题目大意:n个城市,其中k个城市提供面粉,城市之间存在m条路。
现在需要在n个城市中,选择一个城市开面包店;
要求:
1、不能选择有面粉供应的城市;
2、能到达提供面粉的城市;
3、到达提供面粉城市的路径最短;

输入
第一行 n, m and k (1 ≤ n, m ≤ 10^5, 0 ≤ k ≤ n)
接下来m行 每行u, v and l,分别表示城市u和v之间有一条长为l的路 (1 ≤ u, v ≤ n, 1 ≤ l ≤ 109, u ≠ v)
最后一行有k个数字a[i], a[i]表示提供面粉的城市;

输出:
选择的城市;(没有输出-1)

Examples
input
5 4 2
1 2 5
1 2 3
2 3 4
1 4 10
1 5
output
3
input
3 1 1
1 2 3
3
output
-1

代码实现


    int n, m, k;
    cin >> n >> m >> k;
    
    for(int i = 0; i <= n; ++i) g[i].clear();
    
    for (int i = 0; i < m; ++i) {
        lld u, v, l;
        cin >> u >> v >> l;
        g[u].push_back(make_pair(v, l));
        g[v].push_back(make_pair(u, l));
    }
    
    for (int i = 0; i < k; ++i) {
        int tmp;
        scanf("%d", &tmp);
        f[tmp] = 1;
    }
    
    lld ret = -1;
    for (int i = 1; i <= n; ++i) {
        if (f[i]) {
            for (int j = 0; j < g[i].size(); ++j) {
                if (f[g[i][j].first] == 0) {
                    if (ret == -1) {
                        ret = g[i][j].second;
                    }
                    else {
                        ret = min(ret, g[i][j].second);
                    }
                }
            }
        }
    }
    cout << ret;

题目解析
n个点,m条边,k个关键点。在关键点外的集合n-k,中找到一点T,使得T与关键点中任意点连接,并且边长最小。
难度都在读题,容易知道,关键点只要存在一条边与非关键点相连,那就有解。
存下边,把点标记为关键点(f[i]=1)和非关键点(f[i]=0)
遍历查找边由f[i]=1到f[i]=0的最小边即可,无解输出-1。

C.Pythagorean Triples

题目链接
题目大意:给出一个数字n,求一组勾股数中的另外两个,使得三个构成勾股数。

输入:
n (1 ≤ n ≤ 10^9)

输出:
勾股数的另外两个m,k (1 ≤ m, k ≤ 1e18)

Examples
input
3
output
4 5
input
6
output
8 10
input
1
output
-1

代码实现

    lld k, b;
    cin >> k;
    lld mod = 2 - k % 2;
    b = (k * k / mod - mod) / 2;
    if (b == 0) {
        cout << -1;
    }
    else {
        cout << b << " " << b + mod << endl;
    }
    

题目解析
容易知道,n=1,2无解。(最小的勾股数3、4、5)
假设在a^2 + b^2 = c^2 中 令a=n
那么有n * n=c * c-b * b=(c+b) * (c-b)
当n为奇数时,令c-b=1, 有n * n=(b+1+b) => b=(n * n-1)/2
当n为偶数时,令c-b=2, n * n=(b+2+b) * 2 => b=(n * n/2-2)/2
令mod=2-n%2,那么有b= (n * n/mod - mod)/2

D.Persistent Bookcase

题目链接
题目大意:小明有一个书柜,书柜有n层,每层有m个位置可以放书。(书柜一开始是空的)
小明会对书柜进行整理,整理的操作有四种类型:

  • 1 i j,往第i层第j个位置放一本书;(如果上面已经有书则不放)
  • 2 i j,把第i层第j个位置清空;(如果上面没有书则不动)
  • 3 i, 把第i层空位置放上书,把上面有书的位置清空;
  • 4 k,把书柜的状态回到第k次操作后的状态;(如果k=0,表示回到初始状态)

每次操作后输出当前书柜上书的数量。

输入
第一行 n, m and q (1 ≤ n, m ≤ 10^3, 1 ≤ q ≤ 10^5)
接下来q行 每行先输入操作类型,再输入参数。

Examples
input
4 2 6
3 2
2 2 2
3 3
3 2
2 2 2
3 2
output
2
1
3
3
2
4

样例解释

代码实现

struct Node {
    int type, x, y, k;
}node[M];
int flag[N]; // 标志第i行是否翻转
int dfn[M]; // 第i个操作是否已经执行
int a[N][N]; // a[i][j]
int num[N], m; //第i行为1的数量
int sum;    // 当前数量
int ans[M]; // 第i个操作答案
vector<int> g[M]; // 顶点


void lookNext(int k) {
    dfn[k] = 1;
    for (int i = 0; i < g[k].size(); ++i) {
        int u = g[k][i];
        if (dfn[u] == 0) {
            if (node[u].type == 1) {
                if (a[node[u].x][node[u].y] == flag[node[u].x]) {
                    ++sum;
                    ++num[node[u].x];
                    ans[u] = sum;
                    a[node[u].x][node[u].y] = !flag[node[u].x];
                    lookNext(u);
                    a[node[u].x][node[u].y] = flag[node[u].x];
                    --num[node[u].x];
                    --sum;
                }
                else {
                    ans[u] = sum;
                    lookNext(u);
                }
            }
            else if (node[u].type == 2) {
                if (a[node[u].x][node[u].y] != flag[node[u].x]) {
                    --sum;
                    --num[node[u].x];
                    a[node[u].x][node[u].y] = flag[node[u].x];
                    ans[u] = sum;
                    lookNext(u);
                    a[node[u].x][node[u].y] = !flag[node[u].x];
                    ++num[node[u].x];
                    ++sum;
                }
                else {
                    ans[u] = sum;
                    lookNext(u);
                }
            }
            else if (node[u].type == 3) {
                sum = sum - num[node[u].x] + (m - num[node[u].x]);
                num[node[u].x] = m - num[node[u].x];
                flag[node[u].x] = !flag[node[u].x];
                ans[u] = sum;
                lookNext(u);
                flag[node[u].x] = !flag[node[u].x];
                num[node[u].x] = m - num[node[u].x];
                sum = sum + num[node[u].x] - (m - num[node[u].x]);
            }
            else if (node[u].type == 4) {
                ans[u] = sum;
                lookNext(u);
            }
        }
    }
}

   

题目解析
操作1、2比较简单,操作3是组操作,设置flag[i]表示第i行在最终结算时是否翻转,那么有
操作1为a[i][j] = !flag[i].
操作2为a[i][j] = flag[i].
操作3为flag[i] = !flag[i].

操作4较为复杂,回到操作k,k为之前的操作。
考虑到题目对k没有限制,那么k可以为之前的某个回退操作,从而产生递归回退的效果;
同时回退到操作i之后,下一步可以再回退到操作j,这样线性的操作不可取。
但是单个操作只会是一个线性的分支,整个操作序列可以形成多个线性的分支,汇总在一起就是一颗树的表现。
对于第i个操作,操作完毕后的状态为j,连一条边从i到j,表示从第i个操作完之后会进入操作j的状态。
那么对操作1、2、3,i会连上一条边到i+1;操作4,i会连上一条边到k。
对于某一个操作,先执行,然后dfs,最后撤销执行即可。

E. Garlands

题目链接
题目大意:输入k条链,链上的节点在n * m的矩阵上,如图:

每条链有len[i]个点,每个点的输入包括x、y、w表示在n * m矩阵上的坐标和权值。

有q次操作:

  • 操作1,把链上的点翻转(权值由w变成0,或者从0变成w);
  • 操作2,询问子矩阵内点的权值;( 操作2最多2000次)
    n,m,k = 2000, q=10^6.

输入:
第一行 n, m and k (1 ≤ n, m, k ≤ 2000)
接下来输入k条链的信息,每条链的输入格式:
1、 第一行 len,表示长度 (1 ≤ len ≤ 2000)
2、 接下来len行,每行 i, j and w 表示坐标i,j 和 权值w (1 ≤ i ≤ n, 1 ≤ j ≤ m, 1 ≤ w ≤ 1e9)
接着是操作信息,首先是 q 表示操作次数 (1 ≤ q ≤ 1e6)
接下来q行,每行表示操作,操作有两种类型:
1、SWITCH i 表示把第i条链进行翻转;
2、ASK x1 y1 x2 y2 (x1, y1)表示左下角,(x2, y2表示右上角;输出对应矩阵的权值和;

Examples
input
4 4 3
5
1 1 2
1 2 3
2 2 1
2 1 4
3 1 7
4
1 3 1
2 3 3
2 4 3
1 4 1
7
4 1 1
4 2 9
3 2 8
3 3 3
4 3 4
4 4 1
3 4 1
2
ASK 2 2 3 3
ASK 1 1 4 4
output
15
52

样例对应的图

代码实现

#include<cstdio>
#include<cmath>
#include<stack>
#include<map>
#include<set>
#include<queue>
#include<deque>
#include<string>
#include<utility>
#include<sstream>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

typedef long long lld;
const int N = 2010, M = 1000100, inf = 10110110;

struct Node {
    int type, x, y, w;
    Node(int x, int y, int w):x(x), y(y), w(w){};
    Node(){};
};
vector<Node> g[N];
lld d[N][N]; // 前i条链对第j个矩阵的贡献
lld flag[N]; // 标志是否关闭

int n, m, k;

inline void fastRead(int &x){
    char t = getchar();
    bool sign = true;
    while(t < '0' || t > '9') {
        if(t == '-') {
            sign = false;
        }
        t = getchar();
    }
    for(x = 0; t >= '0' && t <= '9'; t = getchar()) {
        x = x * 10 + t - '0';
    }
    if(!sign) {
        x = -x;
    }
}

lld tree[N][N];

struct Ask {
    int x1, y1, x2, y2;
    Ask(int x1, int y1, int x2, int y2):x1(x1), y1(y1), x2(x2), y2(y2){};
};
vector<Ask> matrix;
int ask[M];


char ch[10];

inline int lowbit(int x) {
    return x & -x;
}

void tree_add(int x,int y,int w){
    for(int i = x;i <= n; i += lowbit(i))
        for(int j = y; j <= m; j += lowbit(j))
            tree[i][j] += w;
}
lld tree_sum(int x,int y){
    lld sum = 0;
    for(int i = x; i > 0;i -= lowbit(i))
        for(int j = y; j > 0; j -= lowbit(j))
            sum += tree[i][j];
    return sum;
}

int main(int argc, const char * argv[]) {
    // insert code here...
    cin >> n >> m >> k;
    for (int i = 0; i < k; ++i) {
        int len;
        fastRead(len);
        while (len--) {
            int x, y, w;
            fastRead(x);
            fastRead(y);
            fastRead(w);
            g[i].push_back(Node(x, y, w));
        }
    }
    int q;
    cin >> q;
    for (int i = 0; i < q; ++i) {
        scanf("%s", ch);
        if (ch[0] == 'S') {
            fastRead(ask[i]);
        }
        else {
            int x1, y1, x2, y2;
            fastRead(x1);
            fastRead(y1);
            fastRead(x2);
            fastRead(y2);
            matrix.push_back(Ask(x1, y1, x2, y2));
        }
    }
    
    for (int i = 0; i < k; ++i) {
        for (int j = 0; j < g[i].size(); ++j) {
            tree_add(g[i][j].x, g[i][j].y, g[i][j].w);
        }
        for (int j = 0; j < matrix.size(); ++j) {
            d[i][j] = tree_sum(matrix[j].x2, matrix[j].y2) + tree_sum(matrix[j].x1 - 1, matrix[j].y1 - 1) - tree_sum(matrix[j].x1 - 1, matrix[j].y2) - tree_sum(matrix[j].x2, matrix[j].y1 - 1);
        }
    }
    
    int ans = 0;
    for (int i = 0; i < q; ++i) {
        if (ask[i] == 0) {
            lld ret = 0;
            for (int j = 0; j < k; ++j) { // 枚举k条链
                if (flag[j] == 0) {
                    if (j == 0) {
                        ret += d[j][ans];
                    }
                    else {
                        ret += d[j][ans] - d[j - 1][ans];
                    }
                }
            }
            cout << ret << endl;
            ++ans;
        }
        else {
            flag[ask[i] - 1] ^= 1;
        }
    }
    
    return 0;
}

题目解析
先看最暴力的做法,对于每个switch,把点的值Switch;对于每个ask,遍历所有的链得出结果。
优化部分,添加flag,标志每次switch,询问时再进行计算。
每条链复杂度为O(Len),k条链的修改为O(n) * O(m),询问的时间为O(n) * O(m)。
总的复杂度为O(n * m * 2000) * 2=16 * 1e9。( * 2是因为每次询问都要修改一次、求和一次)

接着使用数据结构来优化。
易知,子矩阵求和使用树状数组即可。求和操作可以优化为O(logN) * O(logM)。
同样在询问的时候再来修改权值,那么有修改复杂度为O(logN) * O(logM) * O(N) * O(M)。
求和复杂度可以忽略,总得复杂度为2000 * log2000 * log2000 * 2000 * 2000 = 8 * 10e10。 (虽然理论上q=10e6限制了当矩阵数为2000时,每次询问前的switch操作有限,但是一条链可以很长,对很长的链进行操作即可,所以最后的修改次数我们还是按N * M来计算)
为什么变大了?

因为每次询问前的修改操作变成耗时操作,如果题目每次在询问前都修改所有的值,复杂度会很高。
继续优化。
每次修改的都是同一个值(整条链为0,整条链恢复),那么可以预处理出这个值d[i][j],表示第i条链对第j个子矩阵的贡献。
这样就可以避免每次询问前修改值,使用之前预处理的计算值即可。
复杂度为5 * 2000 * 2000 * log2000 * log2000 = 2billion;
全部为加法,并且题目给出的时限为3s,可行。

总结

  • A题是简单题;
  • B题是简单题;
  • C题是构造题;
  • D题是DFS;
  • E题是数据结构+离线处理;

只要DIV2没有数论题都能AC,做CF算是赶上大学时候的水平。虽然目前的工作用到的机会比较少,但是做做算法训练还是挺好玩的。
只是喜欢玩,就花点时间在上面;因为对未来可能有用,所以希望自己坚持下去。

附加题

题目一:《宝石》

有一串宝石首尾相连,用一个大写字母表示一个宝石;
现在需要从这一串宝石中截取一段宝石,要求这一段宝石包含ABCDE这5种字母;求剩下最多有多少个宝石?
input
AFBCFFDE
output
2
样例解释:
因为宝石是收尾相连的,所以AFBC和DE是相连的,可以截取这一段下来,剩下FF两个宝石,所以答案为2。

容易知道,我们想要截取一段最短的宝石,包含ABCDE5种宝石;
首先解决首尾相连的问题:把字符串复制一遍放在最后,这样就可以表示循环;
问题变成在字符串str中,找到一个最短的,包含ABCDE 5种字符的子串。
因为对ABCDE的顺序没有要求,故而可以用贪心来解决这个问题:
我们对每一种宝石都设置一个Right数组,Right[i]表示第i种宝石最右边的位置;
假设我们找到了这个最短子串,设最左边的节点为i,最右边的节点为j,有[i, j],那么有 i = min_element(Right, Right + 5)。
然后遍历 N*2的字符串,不断更新Right数组和ans即可。

题目二:《袋鼠》

有n个弹簧排成一列,袋鼠起始位置在第一个弹簧;
输入n个数字,代表n个弹簧的力量;
弹簧的力量为5表示可以往后跳最多5个弹簧;
问袋鼠到达第n个弹簧的最小弹跳次数?

dp[i] 表示到达第i个的最小步数; dp[1]=1;
对于第i个数字是a[i],那么有枚举j, 1<= j <= a[i]
d[i+j]=min(d[i+j], d[i]+1);
表示到达i+j的最优解;
复杂度最多可以到O(N*N)。

优先队列优化:
对dp[i], 打包成pair(i, a[i]) 放入优先队列;
这样每次取出来的都是最小步数,然后判断i+a[j]是否大于等于当前位置,是则更新,不是则丢弃这个解,重新在队列里面取;

扩展

《宝石二》

有一串宝石首尾相连,用一个大写字母表示一个宝石,每个宝石有相应价值;
现在需要从这一串宝石中截取一段宝石,要求这一段宝石包含ABCDE这5种字母;求剩下最大价值数?

input
8
AFBCFFDE
1 2 3 4 5 6 7 8

output
11

《袋鼠二》

袋鼠喜欢在弹簧上弹跳;
有n个弹簧排成一列,每个弹簧可以弹到下一个弹簧;
输入n个数字,代表袋鼠对n个弹簧的喜欢值;
袋鼠只喜欢跳到喜欢值大于等于起始位置喜欢值的弹簧;
袋鼠可以在任意弹簧位置起跳;
袋鼠的开心值=起始点的喜欢值*经过的弹簧数;
求袋鼠最大的开心值。

input
5
1 2 3 4 5

output
9

袋鼠在位置1,2,3,4,5起跳的开心值分别为5,8,9,8, 5。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,383评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,522评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,852评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,621评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,741评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,929评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,076评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,803评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,265评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,582评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,716评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,395评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,039评论 3 316
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,798评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,027评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,488评论 2 361
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,612评论 2 350

推荐阅读更多精彩内容