Codeforces 1028E. Restore Array

题目直通车：http://codeforces.com/problemset/problem/1028/E

解法：设原数组为ar[]，求ar中的最大值的下标ins，依次向前遍历一遍，每一个答案值都为前一个答案值+余值（ar[i]），这样可以达到题意的要求

输出为NO的情况就是所有ar元素相等且不为0

麻烦的地方在于0值得特判，代码中有说明

#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<string.h>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include<fstream>
#include<cstdlib>
#include<ctime>
#include<list>
#include<climits>
#include<bitset>
using namespace std;
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define fopen freopen("input.in", "r", stdin);freopen("output.in", "w", stdout);
#define left asfdasdasdfasdfsdfasfsdfasfdas1
#define tan asfdasdasdfasdfasfdfasfsdfasfdas
typedef long long ll;
typedef unsigned int un;
const int desll[][]={{,},{,-},{,},{-,}};
const ll mod=1e9+;
const int maxn=4e5+;
const int maxm=1e6+;
const double eps=1e-;
int m,n;
int ar[maxn];
ll ansl[maxn];
int main()
{
    scanf("%d",&n);
    for(int i=;i<n;i++)scanf("%d",&ar[i]);
    int ins=,mx=;
    for(int i=;i<n;i++){
        if(mx<ar[i]){
            mx=ar[i];
            ins=i;
        }
    }
    int l=,r=n-;
    while(l<n && ar[l]==mx)l++;
    while(r>= && ar[r]==mx)r--;
    if(l==n){
        if(ar[]>){
            printf("NO\n");
            return ;
        }
    }
    if(l> && r<n-){
        ins=r+;
    }
    int i=ins;
    ins=;
    ansl[i]=ar[i];
    if(mx== && l==n)ansl[i]=;
    while(++ins<n){
        int j;
        if(i==)j=n-;
        else j=i-;
        ansl[j]=ansl[i]+ar[j];
        if(ar[j]== && ansl[j]==mx)ansl[j]*=;//特殊处理0
        i=j;
    }
    if((ar[i]== && ansl[i]==ansl[(i+)%n]) || ansl[i]%ansl[(i+)%n]==ar[i]){//前一部分特殊处理0
        printf("YES\n");
        for(int i=;i<n;i++){
            printf("%I64d%c",ansl[i],i!=n-?' ':'\n');
        }
    }
    else printf("NO\n");

    return ;
}