Codeforces Round #696 (Div. 2)D

最新推荐文章于 2022-11-22 23:15:14 发布

Lqingyyyy

最新推荐文章于 2022-11-22 23:15:14 发布

阅读量164

点赞数

分类专栏： c++ 文章标签：算法

本文链接：https://blog.csdn.net/qqqingyi/article/details/112912221

版权

c++ 专栏收录该内容

65 篇文章 5 订阅

订阅专栏

首先的一个思想肯定是想到
如何将所有变成0那就是从前往后遍历
a[2] = a[2] - a[1]
a[3] = a[3] - (a[2] - a[1])
以此类推直到a[n] = 0才算符合条件
所以此时只要从 1 - n - 1中选出一个数和后者相交换然后判断合法性即可
这样的复杂度需要 on2那肯定是不行的
再次我们想到交换 ai 和 ai + 1 的话
1 - i的数都是不变的所以我们可以用前缀和来储存 1 - i的值
再次我们想到如果是保证交换的值改变的话 i + 2 - n这一部分也是不会变的
所以我们也只需要后缀和即可
怎么判断合法性呢我们只要判断 ai - 前面这一部分 + ai + 1 - 后面这一部分 = 0即可

#include<iostream>
#include<vector>

using namespace std;

const int N = 2e5 + 10;

const int INF = 1e9;

bool check(vector<int> v){
	for(int i = 1; i < v.size(); i++){
		if(v[i] < v[i - 1]) return 0;
		v[i] -= v[i - 1];
	}
	if(v.back() == 0) return 1;
	return 0;
}

int main(){
	int t;
	cin >> t;
	while(t--){
		int ok = 0;
		int n;
		cin >> n;
		vector<int>a(n + 17,0),b(n + 17,0),l(n + 17,0),r(n + 17,0);
		for(int i = 1; i <= n; i++){
			cin >> a[i];
			b[i] = a[i];
		}
		
		if(n == 1){
			cout << "NO" << endl;
			continue;
		}
		
		if(check(a)) ok = 1;
		
		swap(a[1],a[2]);
		if(check(a)) ok = 1;
		swap(a[1],a[2]);
		
		swap(a[n - 1],a[n]);
		if(check(a)) ok = 1;
		swap(a[n - 1],a[n]);
		
		for(int i = 1; i <= n; i++){
			l[i] = r[i] = -INF;
		}
		
		for(int i = 1; i <= n; i++){
			if(b[i] < b[i - 1]) break;
			else{
				l[i] = b[i] -= b[i - 1];
			}
		}
		
		for(int i = 1; i <= n; i++){
			b[i] = a[i];
		}
		
		for(int i = n; i >= 1; i--){
			if(b[i] < b[i + 1]) break;
			else {
				r[i] = b[i] -= b[i + 1];
			}
		}
		
		for(int i = 2; i <= n - 2; i++){
			vector<int> t = {l[i - 1],a[i + 1],a[i],r[i + 2]};
			if(l[i - 1] == -INF || r[i + 2] == -INF) continue;
			if(check(t)){
				ok = 1;
				break;
			}
		}
		
		if(ok) cout << "YES" << endl;
		else cout << "NO" << endl;
	}
	
	
	
	return 0;
}