bzoj 5071: [Lydsy十月月赛]小A的数字 乱搞

题意

小A成为了一个数学家，他有一串数字A1,A2…An
每次可以进行如下操作，选择一个数字i(1 < i <= n),将(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.可是他太小，不会算，请你帮帮他
n<=100000,|Ai|<=1000000

分析

比赛的时候没想出来。。。
考虑前缀和s[i]的变化，会发现选择i进行操作实际上就是交换了s[i]和s[i-1]，那么只要把A和B的前缀和求出来，排序后判是否相等即可。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;const int N=100005;int n,a[N],b[N];int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}bool check(){    sort(a+1,a+n+1);sort(b+1,b+n+1);    for (int i=1;i<=n;i++) if (a[i]!=b[i]) return 0;    return 1;}int main(){    int T=read();    while (T--)    {        n=read();        for (int i=1;i<=n;i++) a[i]=read(),a[i]+=a[i-1];        for (int i=1;i<=n;i++) b[i]=read(),b[i]+=b[i-1];        if (check()) puts("YES");        else puts("NO");    }    return 0;}