Codeforces.1110E.Magic Stones(思路 差分)

题目链接

---

听dalao说很nb，做做看（然而不小心知道题解了）。

(Description)

给定长为(n)的序列(A_i)和(B_i)。你可以进行任意多次操作，每次操作任选一个(iin[2,n-1])，把(A_i)变成(A_{i-1}+A_{i+1}-A_i)。
求能否将序列(A_i)变成(B_i)。
(nleq10^5)。

(Solution)

(A_i o A_{i-1}+A_{i+1}-A_i)这个形式很有趣，求出(A_i)的差分序列(d_i)试试看。
令(d_{i-1}=A_i-A_{i-1}, d_i=A_{i+1}-A_i)。
如果变换(A_i)，那么(d_{i-1})会变成(A_{i+1}-A_i=d_i)，(d_i)会变成(A_i-A_{i-1}=d_{i-1})。
也就是我们可以任意交换(A_i)的差分序列(d_i)。
所以再求出(B_i)的差分序列(d_i')，判断(d_i)能否变成(d_i')，再判一下(A_1)是否等于(B_1)即可。

---

//46ms	900KB
#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
typedef long long LL;
const int N=1e5+5;

int d1[N],d2[N];

inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-48,c=gc());
	return now;
}
inline bool Check(const int n)
{
	for(int i=1; i<n; ++i) if(d1[i]!=d2[i]) return 0;
	return 1;
}

int main()
{
	const int n=read();
	int A1=read();
	for(int i=1,pre=A1,now; i<n; ++i) d1[i]=(now=read())-pre, pre=now;
	int B1=read();
	for(int i=1,pre=B1,now; i<n; ++i) d2[i]=(now=read())-pre, pre=now;
	std::sort(d1+1,d1+n), std::sort(d2+1,d2+n);
	puts(A1==B1&&Check(n)?"Yes":"No");

	return 0;
}