Codeforces 1028E Restore Array 构造

我发现我构造题真的不会写， 想了好久才想出来。。

我们先把n = 2， 所有数字相等， 所有数字等于0的都特判掉。

找到一个b[ i ] > b[ i - 1 ]的位置把它移到最后一个位置， 并且使其a[ i ] = b[ i ]然后从后往前构造。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PLL pair<LL, LL>
#define PLI pair<LL, int>
#define PII pair<int, int>
#define SZ(x) ((int)x.size())
#define ull unsigned long long
 
using namespace std;
 
const int N = 1e6 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + ;
const double eps = 1e-;
const double PI = acos(-);
 
int n, who = -, id[N];
LL a[N], b[N];
bool flag;
int main() {
    scanf("%d", &n);
    for(int i = ; i < n; i++) {
        scanf("%lld", &b[i]);
        flag |= b[i];
    }
    for(int i = ; i < n; i++) {
        if(b[(i-+n)%n] < b[i]) who = i;
    }
    if(!flag) {
        puts("YES");
        for(int i = ; i < n; i++) printf("1 ");
        puts("");
    } else if(n == ) {
        if(a[] == a[]) puts("NO");
        else {
            puts("YES");
            if(a[] > a[]) printf("%lld %lld\n", a[], a[] + a[]);
            else printf("%lld %lld\n", a[] + a[], a[]);
        }
    } else {
        if(who == -) return puts("NO"), ;
        int now = , cur = (who + ) % n;
        for(int i = ; i <= n; i++) {
            id[now++] = cur;
            cur = (cur + ) % n;
        }
        a[id[n - ]] = b[id[n - ]];
        for(int i = n - ; i >= ; i--) {
            LL low = b[id[(i -  + n) % n]] + ;
            LL mo = a[id[i + ]];
            a[id[i]] = low + (((b[id[i]] % mo) - (low % mo)) % mo + mo) % mo;
        }
        puts("YES");
        for(int i = ; i < n; i++) printf("%lld ", a[i]);
        puts("");
    }
    return ;
}
 
/*
*/