Codeforces Round #696 (Div. 2) D. Cleaning(思维+前缀+枚举）

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
#include
#include
#include
#include
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 ;
}