codeforces1028E: Restore Array

小清新构造题。

选择一个位置$b_{i-1}<b_i且b_i=max_b$，然后把这个位置循环位移到$n$。

然后构造$a_n=b_n$。

$a_i=sum_{j=i}^{n-1}b_j+2*b_n$。

带回去检验一下，发现是对的。

思路大概就是往$a_i-a_{i+1}=b_i$的方向想吧。

#include <bits/stdc++.h>
using namespace std;
#define M 150000
#define LL long long
#define Next(x) x % n + 1
#define pre(x) (x == 1 ? n : x - 1)
int b[M], c[M];
LL a[M];
int n;
int main() {
    scanf("%d", &n);
    for(int i = 1; i <= n; ++ i) {
        scanf("%d", &b[i]);
    }
    b[0] = -1;
    int mx = 0;
    for(int i = 1; i <= n; ++ i) {
        if(b[i] > b[mx] && b[pre(i)] < b[i]) {
            mx = i;
        }
    }
    if(mx == 0) {
        if(b[1] != 0) puts("NO");
        else {
            puts("YES");
            for(int i = 1; i <= n; ++ i) {
                printf("1 ");
            }
            puts("");
        }
        return 0;
    }
    int st;
    mx = Next(mx);
    for(int i = 1; i <= n; ++ i, mx = Next(mx)) {
        if(mx == 1) st = i;
        c[i] = b[mx];
    }
    a[n] = c[n];
    a[n - 1] = 2 * c[n] + c[n - 1];
    for(int i = n - 2; i >= 1; -- i) {
        a[i] = a[i + 1] + c[i];
    }
    puts("YES");
    for(int i = st, j = 1; j <= n; ++ j, i = Next(i)) {
        printf("%lld ", a[i]);
    }
}