Codeforces Round #696 (Div. 2) D. Cleaning(思维+前缀+枚举)
链接 :
题意 :
给出一段序列ai ,每次可以选择相邻的两个都不为0的ai 与 ai+1 令其都− 1 ,(这个操作可使用无限次)。在操作之前,你可以使用一次特权:交换相邻的两个数的位置(只能使用一次)。问是否可以将序列全部变为0 . 可以输出YES , 否则输出NO。
思路 :
首先要满足队首一定要小于等于后面的值 , 否则一定不成立。然后维护一个前缀 ,
如果a[i] >= pre[i-1] 则 pre[i] -= pre[i-1] 在这里 ,pre[i-1]是[i-1]经过相减后的值 , 否则pre[i] = inf , 标记为不能成立 ; 后缀同样如此 , 但是后缀标记不成立时要注意不能和前缀相同 , 即 suf[i] = inf / 2 , 否则无法区分 。
之后再枚举交换的两个数 。
代码 :
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std ;
const int maxn = 2e5 + 10 ;
typedef long long ll ;
ll a[maxn] , pre[maxn] , suf[maxn] ;
const ll inf = 1e18 ;
int main(){
ios::sync_with_stdio(false) ;
cin.tie(0) ;
cout.tie(0) ;
int t ;
cin >> t ;
while(t--){
ll n ;
cin >> n ;
memset(pre , 0 , sizeof(pre)) ;
memset(suf , 0 , sizeof(suf)) ;
for(ll i = 1 ; i <= n ; i++) cin >> a[i] ;
for(ll i = 1 ; i <= n ; i++){ //前缀
if(a[i] >= pre[i-1]) pre[i] = a[i] - pre[i-1] ;
else pre[i] = inf / 2 ;
}
for(ll i = n ; i >= 1 ; i--){ //后缀
if(a[i] >= suf[i+1]) suf[i] = a[i] - suf[i+1] ;
else suf[i] = inf ;
}
bool flag = 1 ;
for(ll i = 1 ; i <= n - 1 ; i++){
if(pre[i] == suf[i+1]){
flag = 0 ;
break ;
}
else{ //尝试交换
ll tmp1 = a[i] ;
ll tmp2 = a[i+1] ;
if(tmp1 >= suf[i+2] && tmp2 >= pre[i-1]){
tmp1 -= suf[i+2] ;
tmp2 -= pre[i-1] ;
ll cnt = tmp1 - tmp2 ;
if(cnt == 0){
flag = 0 ;
break ;
}
}
}
}
if(!flag) cout << "YES" << endl ;
else cout << "NO" << endl ;
}
return 0 ;
}