codeforces 1028E Restore Array(构造)

题目:给一个长度为n的数组a,ai=b[(i+1)%n]%b[i]  ,判断存不存在满足条件的b数组,有的话输出任意一个符合条件的b。

思路:一开始发现,找到最大的数a[k]而且a[k]前一个比它小,让b[k]=a[k]然后往前不断加a[i]并得到b[i],可以得到符合答案的b。事实证明wa了。。。把一开始找到的a[k],给它乘以一个特别大的系数就好了。全是0的时候要特殊判断。

#include <bits/stdc++.h>
using namespace std;
#define inf 1e9
typedef long long ll;
const int maxn=150000;
int n;
ll a[maxn],b[maxn];
int main()
{
    scanf("%d",&n);
    int k=-1,z=0;
    ll ma=-1,mi=inf;
    for(int i=0;i<n;i++)
        scanf("%lld",&a[i]);
    for(int i=0;i<n;i++)
    {
        if(a[i]>0) z=1;
        if(a[i]>ma&&a[i]>a[(i-1+n)%n])
        {
            k=i;
            ma=a[i];
        }
        mi=min(mi,a[i]);
    }
    if(z==0)
    {
        puts("YES");
        for(int i=0;i<n;i++)
            printf("1 ");
        return 0;
    }
    if(k==-1||mi==ma)
    {
        puts("NO");
        return 0;
    }

    b[k]=a[k];
    k=(k-1+n)%n;
    b[k]=b[(k+1)%n]*10000000+a[k];
    for(int i=1;i<n-1;i++)
    {
        int j=(k-i+n)%n;
        b[j]=a[j]+b[(j+1)%n];
    }
    puts("YES");
    for(int i=0;i<n;i++)
        printf("%lld ",b[i]);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/dllpXFire/article/details/82154015