BZOJ 5071 [Lydsy 十月月赛] 排序 解题报告

Description

小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.可是他太小，不会算，请你帮帮他

【解题报告】

$(ai-1,ai,ai+1) => (ai-1+ai,-ai,ai+1+ai)$
我们考虑其前缀和变化：
设$x$为$sumi-2$
在变换前：
$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$交换了一下。
比较前缀和就可以了

代码如下：

/**************************************************************
    Problem: 5071
    User: onepointo
    Language: C++
    Result: Accepted
    Time:4584 ms
    Memory:1604 kb
****************************************************************/

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010

int t,n;
int a[N],b[N];

int main()
{
    for(scanf("%d",&t);t;--t)
    {
        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+n+1);sort(b+1,b+n+1);bool flag=1;
        for(int i=1;i<=n;++i)if(a[i]!=b[i]){flag=0;break;}
        puts(flag?"YES":"NO");
    }
    return 0;
}