CF 1141-C. Polycarp Restores Permutation

题目: 传送门
思路: 因为每两个数之间的差值已经确定,由差分思想我们知道只要给定一个序列的第一个数,后面的每个数都能求出来(并且唯一),我们不妨就先假设为 1 ,求出 整个 序列,找到最小值 ,然后让所有数加上该最小值变为 1 的值 , 这时的序列是满足从 1 开始的条件的,我们只需检验是否是 1~n之间的数是否都出现且出现过一次即可。
代码如下:

#include <iostream>
#include <algorithm>
#include <set>
using namespace std;

int n;
long long a[200010];
set<long long> s;

int main() {
    cin>>n;
    a[0] = 0;
    long long mins=0;
    for(int i = 1;i<=n-1;i++) {
        int d;
        cin>>d;
        a[i]=a[i-1]+d;
        mins= min(mins,a[i]);
    }
    long long d= 1-mins;
    for(int i=0;i<n;i++) {
        a[i] +=d;
        if(a[i]<=n) s.insert(a[i]);
    }
    if(s.size()==n) {
        for(int i=0;i<n;i++) {
            cout<<a[i];
            if(i==n-1) cout<<endl;
            else cout<<' ';
        }
    }
    else {
        cout<<"-1"<<endl;
    }
    return 0;
}


猜你喜欢

转载自blog.csdn.net/qq_43305984/article/details/89206648
今日推荐