【Codeforces Round #547 (Div. 3) C. Polycarp Restores Permutation】思维题-CSDN博客

本文链接：https://blog.csdn.net/qq_38891827/article/details/88696996

C. Polycarp Restores Permutation

题意

给你一个长度为 $n$ 的数组差分后的数组(差分数组长度为 $n - 1$ )。
问原数组是否是一个 $1 - n$ 的排列。

做法

首先我们可以知道如果固定数组中的第一个数，就能还原数组，而且如果原数组是 $[1, n]$ 排列，还原后的数组一定是 $[k, k + n - 1]$ 的排列，所以只要判断还原后的数组是否满足情况即可。注意下标不要越界。

代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 2e5+5;
const int INF = 0x3f3f3f3f;
int a[maxn];
int vis[maxn];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n-1;i++) scanf("%d",&a[i]);
    int minn=INF,maxx=-INF;
    a[0]=0;
    for(int i=1;i<=n-1;i++)
    {
        a[i]=a[i-1]+a[i];
        minn=min(minn,a[i]);
        maxx=max(maxx,a[i]);
    }
    minn=min(minn,0);
    maxx=max(maxx,0);
    if(maxx-minn+1==n)
    {
        for(int i=0;i<=n-1;i++) a[i]=a[i]+(1-minn);
        for(int i=0;i<=n-1;i++)
        {
            if(vis[a[i]])
            {
                printf("-1\n");
                return 0;
            }
            vis[a[i]]=1;
        }
        for(int i=0;i<=n-1;i++) printf("%d ",a[i]);
    }
    else printf("-1\n");
    return 0;
}