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;
}