Polycarp Restores Permutation
【题意分析】题意大概是给定一个串,包含从1到n所有的数字。但是给定的是相邻数字的差,需要复原这个串。
大概分析以后发现给定的是一个差分数组,所以只需要枚举第一个元素就可以确定所有元素的值。
问题是如何判断可行,一个一个判断显然是不行的——会超时。
剪枝技巧1:在得到差分数组的时候记录最小值和最大值,a[1]+最小值肯定是1,a[1]+最大值肯定是n,否则肯定不是答案
剪枝技巧2:可能会出现重复的情况,重复的原因是差分数组含有相同的元素,因此先将差分数组排个序,然后检查是否含有相同的元素,若含有肯定是不存在的,若不含有,肯定是存在的——a[1]+q[i]在1到n之间,而且互不相同,而且含有n个,肯定就是答案了。
代码如下:(这么简单的程序我wa了7次,自己真的是太菜了,也有一方面原因是后来做题太少)
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<climits>
using namespace std;int n;
const int MAXN=200005;
int q[MAXN],p[MAXN],t[MAXN];
bool check[MAXN];
int minn=INT_MAX,maxn=INT_MIN;
int tmp,flag;void work(int x)
{memset(check,0,sizeof(check));p[1]=x;for(int i=2;i<=n;i++){p[i]=x+q[i];}
}int main()
{scanf("%d",&n);//memset(q,0,sizeof(q));q[1]=0; t[1]=0;for(int i=2;i<=n;i++){scanf("%d",&tmp);q[i]=q[i-1]+tmp;t[i]=q[i];if(q[i]<minn) minn=q[i];if(q[i]>maxn) maxn=q[i];}sort(t+1,t+n+1);flag=1;for(int i=2;i<=n;i++){if(t[i]==t[i-1]){flag=0;break;}}if(flag==0){printf("-1");return 0;}//memset(p,0,sizeof(p));flag=0;for(int i=1;i<=n;i++){if(i+minn<1) continue;if(i+maxn>n) continue;work(i);flag=1;for(int i=1;i<n;i++){printf("%d ",p[i]);}printf("%d",p[n]);break;}if(flag==0){printf("-1");}return 0;
}