题意是给你一个数组 有一个操作 是对于 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;
}