Codeforces Round #696 (Div. 2) D. Cleaning

【DP+?】 Cleaning

题面：Codeforces Round #696 (Div. 2) D. Cleaning
题意：给出n(n<=2e5)堆石子，每堆石子个数1~1e9，每次操作可以使相邻两堆同时减1个，问是否能将每个堆都减为0个，但是在操作前有一次机会，可以选择两堆相邻石子进行交换（也可以不交换）。

思路：
首先，第一堆的石子必须由第二堆减去，他们的差必须由第三堆减去，以此类推，最终如果最后一堆个数能到0，那么就可以。同理，从后往前也可以判断一次。
然后考虑交换的操作，我们可以假设交换的石子为p和p+1，那么对p前面的石子和对p+1后面的石子是没有影响的，因此我们可以预处理出一个pre[i]，表示前面的石子减到i位置后剩下的个数，再预处理一个suf[i]，表示后面的石子减到i位置后剩下的个数。
最后枚举交换的位置p，然后判断一下 pre[p-1],a[p],a[p+1],suf[p+2] 这四堆的石子是否能减为0 或者 pre[p-1],a[p+1],a[p],suf[p+2] 这交换后的四堆是否能减为0，就结束了。

#include<bits/stdc++.h>
#define ll long long
#define mem(b,t) memset(b,t,sizeof(b))
#define INF 0x3f3f3f3f
#define rep(i,n,m) for(int i=n;i<=m;i++)
#define per(i,n,m) for(int i=n;i>=m;i--)
#define bug1(i) cout<<i<<endl
#define bug2(i,j) cout<<i<<" "<<j<<endl
#define mod 1000000007
#define maxn 200005
#define eps 1e-6
#define pi acos(-1)
using namespace std;
ll a[maxn],pre[maxn],suf[maxn];
int check(ll a,ll b,ll c,ll d){
	if(a<0||d<0) return 0;
	if(a>b) return 0;
	else b-=a;
	if(b>c) return 0;
	else c-=b;
	if(c>d) return 0;
	else d-=c;
	if(d==0) return 1;
	return 0;
}
int main(){
	int t,n;
	cin>>t;
	while(t--){
		cin>>n;
		mem(pre,0);
		mem(suf,0);
		rep(i,1,n) scanf("%lld",&a[i]);
		rep(i,1,n){
			if(pre[i-1]>=0 && a[i]-pre[i-1]>=0)
			pre[i]=a[i]-pre[i-1];
			else pre[i]=-1;
		}
		per(i,n,1){
			if(suf[i+1]>=0 && a[i]-suf[i+1]>=0)
			suf[i]=a[i]-suf[i+1];
			else suf[i]=-1;
		}
		//rep(i,1,n){
//			cout<<pre[i]<<" ";
//		}
//		cout<<endl;
//		rep(i,1,n){
//			cout<<suf[i]<<" ";
//		}
//		cout<<endl;
		int f=0;
		rep(i,1,n-1){
			if(check(pre[i-1],a[i],a[i+1],suf[i+2]) || check(pre[i-1],a[i+1],a[i],suf[i+2])){
				puts("YES");
				f=1;
				break;
			}
		}
		if(!f) puts("NO");
	}
	return 0;
}