AtCoder Regular Contest 129 D题

题目

题意是给你一个数组 有一个操作 是对于 i - 1 i i + 1 分别 -1 + 2 -1 若选 i == n i + 1 = 1
首先我们就知道一个思路 -1 +2 -1 是 很明显的 一个差分数组后的一个方案 但是我有点nt 我进行了两次前缀和 使得最后得题意转化成了 给你个数组 每次操作可以让其中一个数 -1 最后得数组要变为递增 所以。。完全搞不来了 但是只前缀和一次的话 就是个经典问题 每次 操作 -1 +1 使得数组变为0 那就只要进行两次循环 暴力即可

#include<iostream>
#define int long long

using namespace std;

const int N = 2e5 + 10;

int a[N],b[N];

signed main(){
    int n;
    cin >> n;
    int s = 0;
    for(int i = 1; i <= n; i++) scanf("%lld",&a[i]),s += a[i];

    if(s != 0){
        cout << -1 << endl;
        return 0;
    }

    for(int i = 1; i <= n; i++) b[i] = b[i - 1] + a[i - 1];
    s = 0;
    for(int i = 1; i <= n; i++) s += b[i];
    if(s % n){
        cout << -1 << endl;
        return 0;
    }

    s /= n;

    for(int i = 1; i <= n; i++) b[i] -= s;

    int sum = 0,d;
    for(int i = 1; i <= n; i++){
        if(i == n) d = 1;
        else d = i + 1;
        if(b[i] > 0) sum += b[i],b[d] += b[i],b[i] = 0;
    }
    for(int i = 1; i <= n; i++){
        if(i == n) d = 1;
        else d = i + 1;
        if(b[i] > 0) sum += b[i],b[d] += b[i],b[i] = 0;
    }

    for(int i = 1; i <= n; i++){
        if(b[i] != 0){
            cout << -1 << endl;
            exit(0);
        }
    }

    cout << sum << endl;







    return 0;
}