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