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