题意:
给一个序列,你有一个操作且只能操作一次,就是交换两个相邻的数字,然后如果序列中可以实现无限次的相邻的数字减一,如果可以最终序列全为0则输出yes 否则no
思路 : 由于如果有空的情况就是a1,0,a2。。这种的,a1与a2不相邻,所以一定是no 那么我们只有两种删除的方向,一个从开始到最后,一个从最后到开始。
进一步思考 那么如果有后一个数字大于前一个数字,我们从前往后就会出现后一个为0,前一个还不为0的情况,所以一定不可能。同理从后往前删除是一样的情况
a1 = sum
a2-a1 = sum
a3-(a2-a1) = sum
a4-(a3-(a2-a1)) = sum
此时每个sum一定是大于等于零的,才能实现后减前一个,否则一定不可以
首先想暴力,然后超时了
所以我们必须预处理出来一个东西,可以让交换后判断的时间大大缩减,所以我们预先从前往后求出pre数组 从后往前求出 suf 数组。
(重点)如果pre数组与suf数组可以接上,那么代表可以输出yes 这里是整个解法的核心。
具体代码有解释
#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(){
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 ; // 预处理pre,当有不符合情况的时候,pre数组从此往后全为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 ;// 从后往前,如果有不符合,那么全为inf
}
bool flag = 1 ;
for(ll i = 1 ; i <= n - 1 ; i++)
{
if(pre[i] == suf[i+1]){ // 如果可以接上的话直接输出yes就行
flag = 0 ;
break ;
}
else
{
ll tmp1 = a[i] ; // 如果接不上,那么交换a[i]与a[i+1] 然后实现pre[i] 和 suf[i+1] 如果接的上输出yes
ll tmp2 = a[i+1] ;
if(tmp1 >= suf[i+2] && tmp2 >= pre[i-1])
{
tmp1 -= suf[i+2] ;
tmp2 -= pre[i-1] ;
if(tmp1 == tmp2)
{
flag = 0 ;
break ;
}
}
}
}
if(!flag) cout << "YES" << endl ;
else cout << "NO" << endl ;
}
return 0 ;
}