A. Directional Increase -前缀和与差分理解 + 思维

最新推荐文章于 2024-07-12 18:56:03 发布

.y.a.o.

最新推荐文章于 2024-07-12 18:56:03 发布

阅读量112

点赞数 1

文章标签：算法 c++ 前缀和差分

本文链接：https://blog.csdn.net/m0_75129175/article/details/134142030

版权

题面

分析

观察指针移动的性质，可以发现每一段都是从起点走到终点，在原路返回，这样每一段也就表示，在起点处加一，在终点处减一，形成了很明显的差分结构，思考能否构造出a数组的关键就是他的前缀和数组b的b[n]必须等于0，并且每一个 $b_i$ 都不能小于0，因为起点大于0，终点小于0，所有数都应该是大于等于0的，在某一个下标一旦前缀和数组元素等于0，代表开始原路返回，这是走过最长的一段，那么后面所有的前缀和元素都必须是0.

代码

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

void solve() {
    int n;
    cin >> n;
    vector<ll> a(n + 1);
    for(int i = 1; i <= n; i ++) {
        cin >> a[i];
        a[i] += a[i - 1];
    }
    if(a[n] != 0) {
        cout << "No\n";
        return ;
    }
    int flag = 0;
    for(int i = 1; i <= n; i ++) {
        if(a[i] < 0) {
            flag = 1;
            break;
        }
    }
    int cnt = 0;
    for(int i = 1; i <= n; i ++) {
        if(a[i] == 0) cnt = 1;
        else {
            if(cnt == 1) {
                flag = 1;
                break;
            }
        }
    }
    if(flag == 1) cout << "No\n";
    else cout << "Yes\n";
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T;
    cin >> T;
    while(T --) {
        solve();
    }
}