D. Boris and His Amazing Haircut

Problem - D - Codeforces

问题描述:剪发,将数组a减为数组b,有m个剪刀,每个剪刀只可以用一次且可以在任意区间内剪发,将长度大于mi的减为mi。现在有m数组,数组元素是第i个剪刀可以剪到mi,问能否将a减为b。

洛谷翻译:

D. Boris and His Amazing Haircut_第1张图片

思路:一定是先减最长的,再减短的。在减的时候会将这个a数组渐渐减成多个数组,再对这些数组进行这些操作,判断给出的m数组是否满足可以进行这些操作。

如果是b[0],因为是第一个,所以一定需要一个剪刀m = b[0]

b[1]时,有三种情况:

  • b[1] < b[0]:因为接下来要减的少,所以也要用一个剪刀m = b[1]
  • b[1] = b[0]:相同,上一个可以被覆盖,不需要额外操作。
  • b[1] > b[0]:这是,由于b[1] > b[0],前面大于b[1]的剪刀都不可以用。因为如果用了,那么a[1]就会小于b[1],此时不满足条件。

发现此时具有单调栈性质:通过从0到n-1进行遍历b数组,先将栈中小于bi的出栈,之后判断是否为空或者已经存在栈中(sk.top() == bi),如果为空或者bi不在栈中,入栈,表示一定需要这个

代码:

void solve() {
    int n; cin>>n;
    vector<int> a(n), b(n);
    for(auto &t: a) cin>>t;
    for(auto &t: b) cin>>t;
    bool ok = true;
    for(int i = 0; i < n; ++i) if(a[i] < b[i]) ok = false;
    map<int,int> mii;
    int m; cin>>m;
    for(int i = 0; i < m; ++i) {
        int t; cin>>t;
        mii[t]++;
    }
    stack<int> sk;
    for(int i = 0; i < n; ++i) {
        while(sk.size() && sk.top() < b[i]) sk.pop();
        if(a[i] == b[i]) continue;
        if(sk.empty() || sk.top() != b[i]) {
            sk.push(b[i]);
            mii[b[i]]--;
        }
    }
    for(auto t: mii) {
        ok &= t.vs >= 0;
    }
    puts(ok ? "YES" : "NO");
}

你可能感兴趣的:(算法题,cf,算法,数据结构)