BZOJ5071[Lydsy十月月赛] 小A的数字 解题报告

Problem Statement
小A成为了一个数学家，他有一串数字A1,A2...An每次可以进行如下操作，选择一个数字i,将(Ai−1,Ai,Ai+1)
变为(Ai−1+Ai,−Ai,Ai+1+Ai),特别地，若i=N,则(An−1,An)变为(An−1+An,−An).小A很好奇，能否通过若干次操作，得到他的幸运数列B1,B2...Bn.可是他太小，不会算，请你帮帮他
解题报告
我们设x=sumi−2，sumi表示前i位的前缀和。
sumi−1=x+ai−1;
sumi=x+ai−1+ai;
sumi+1=x+ai−1+ai+ai+1;
经过题目所说的变换后：
sumi−1=x+ai−1+ai;
sumi=x+ai−1;
sumi+1=x+ai−1+ai+ai+1;
可见，我们的一次变换只是把sumi−1和sumi换了一个位置。那么我们可以把A,B两个数组排序，比较他们的前缀和是否相同就好了。
代码如下：

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int N=1e5;int T,n;int a[N+5],b[N+5];int main(){    for(scanf("%d",&T);T;T--)    {        bool flag=false;        scanf("%d",&n);        for(int i=1;i<=n;i++)scanf("%d",&a[i]),a[i]+=a[i-1];        for(int i=1;i<=n;i++)scanf("%d",&b[i]),b[i]+=b[i-1];        sort(a+1,a+1+n);sort(b+1,b+1+n);        for(int i=1;i<=n;i++)if(a[i]!=b[i]){flag=true;break;}        printf("%s\n",flag?"NO":"YES");    }    return 0;}