CodeForces - 1474D Cleaning（前缀和+后缀和+思维）

题目链接：点击这里

题目大意：
有 $n$堆石子，第 $i$ 堆石子有 $a_i$ 个，每次操作可以选取两个相邻的且数量不为 $0$ 的石子堆里同时各拿走一个石子。数量为 $0$ 的石子堆不会消失。在所有操作之前可以交换两个相邻的石子堆，问最终是否能移走所有的石子

题目分析：
在不交换相邻元素的情况下我们可以 $O(n)$ 扫一遍序列来判断是否有解
如果暴力枚举交换位置就是先枚举 $1$ 到 $i-1$ 操作的结果 $x$， $i+2$ 到 $n$ 操作的结果 $y$，再判断剩余的四个元素构成的序列 $x,a_{i+1},a_i,y$ 是否有解，时间复杂度为 $O(n^2)$
因为每次都是让一段连续的区间交换，所以我们可以用一个前缀数组 $pre$ 来处理并记录前 $i$ 堆石子合并的结果，进而可以 $O(1)$ 查询出 $1$ 到 $i-1$ 操作的结果；后缀同理
这样时间复杂度就降到 $O(n)$ 了

具体细节见代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
ll read()
{
	ll res = 0,flag = 1;
	char ch = getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch == '-') flag = -1;
		ch = getchar();
	}
	while(ch>='0' && ch<='9')
	{
		res = (res<<3)+(res<<1)+(ch^48);//res*10+ch-'0';
		ch = getchar();
	}
	return res*flag;
}
const int maxn = 2e5+5;
const int mod = 1e9+7;
const double pi = acos(-1);
const double eps = 1e-8;
ll n,a[maxn],pre[maxn],suf[maxn];
bool flag;
int main()
{
	int t = read();
	while(t--)
	{
		n = read();
		for(int i = 1;i <= n;i++)
		{
			a[i] = read();
			if(pre[i-1] == -1) pre[i] = -1;
			else pre[i] = max(-1LL,a[i]-pre[i-1]);
		}
		suf[n+1] = 0;
		for(int i = n;i;i--)
		{
			if(suf[i+1] == -1) suf[i] = -1;
			else suf[i] = max(-1LL,a[i]-suf[i+1]);
		}
		flag = pre[n] ? false : true;
		for(int i = 1;i < n && flag;i++) if(pre[i] == -1) flag = false;
		if(flag)
		{
			puts("YES");
			continue;
		}
		for(int i = 1;i < n && !flag;i++)
		{
			if(pre[i-1]==-1 || suf[i+2]==-1) continue;
			ll tmp[4] = {pre[i-1],a[i+1],a[i],suf[i+2]};
			for(int j = 1;j < 4;j++)
				if(tmp[j-1] == -1) tmp[j] = -1;
				else tmp[j] = max(-1LL,tmp[j]-tmp[j-1]);
			flag |= (tmp[0]>=0 && tmp[1]>=0 && tmp[2]>=0 && !tmp[3]); 
		}
		if(flag) puts("YES");
		else puts("NO");
	}
	return 0;
}