每日一题 - 240119 - D. Berserk Monsters-CSDN博客

本文链接：https://blog.csdn.net/CodeToLiao/article/details/135708078

本文介绍了如何使用C++STLvector实现一个动态数组问题，通过观察和处理数组中元素的变化，找出规律并更新数组范围，涉及到时间复杂度为O(N)。过程中展示了如何利用STL容器和迭代器进行高效操作。

摘要由CSDN通过智能技术生成

D. Berserk Monsters

TAG - $算法 - 【 ST L - v ec t or 】、思维 - 【找规律、全局退化成局部】$
时间复杂度 - $O (N)$

//
#include <bits/stdc++.h>
using namespace std;
// #define int long long

void solve() {
    int n;
    scanf("%d", &n);

    vector<int> a(n + 1), d(n + 1), L(n + 1), R(n + 1);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    for (int i = 1; i <= n; i++) scanf("%d", &d[i]);

    for (int i = 1; i <= n; i++) {
        L[i] = i - 1;
        R[i] = i + 1;
    }
    L[1] = R[n] = -1;

    vector<int> v;
    for (int i = 1; i <= n; i++) v.push_back(i);

    vector<int> used(n + 1);
    for (int i = 1; i <= n; i++) {
        vector<int> die;

        // 有变化的部分
        for (auto j : v) {
            int tt = (L[j] != -1 ? a[L[j]] : 0) + (R[j] != -1 ? a[R[j]] : 0);
        
            if (tt > d[j]) die.push_back(j);
        }
        printf("%d%c", die.size(), " \n"[i == n]);

        // cout << "die: ";
        // for (auto j : die) cout << j << " ";
        // cout << "@" << endl;

        for (auto j : die) used[j] = i;     // 打上标记 防止加入被淘汰的怪兽

        v.clear();
        for (auto j : die) {    // 只加入有（可能）变化的部分
            if (L[j] != -1) {
                R[L[j]] = R[j];
                if (used[L[j]] < i) {
                    v.push_back(L[j]);
                    used[L[j]] = i;
                }
            }
            if (R[j] != -1) {
                L[R[j]] = L[j];
                if (used[R[j]] < i) {
                    v.push_back(R[j]);
                    used[R[j]] = i;
                }
            }
        }

        // cout << "v: ";
        // for (auto j : v) cout << j << " ";
        // cout << "@" << endl;
    }
}

signed main() {
    int t = 1;
    scanf("%d", &t);
    while (t--) solve();

    return 0;
}