D. Cleaning
题意
思路
这个题刚开始写没有太多的思路,最后看了题解才知道怎么写。
你首先需要自己发现的性质就是,第一个元素想要变成0,必须要第二个元素大于第一个元素,然后第三个元素应该大于(第二个元素减去第一个元素),从后往前也是一样的。
我们可以预处理一个pre[i],里面存着a[i]减去前面的已经处理的差值。
就比如 pre[3] 就等于 a[3]-(a[2]-a[1])也等于 a[i]-pre[i]。
一样的 也可以预处理一个 suf[i].
这样我们就可以O(1)的时间判断这个数组是否满足题目中的条件。
满足条件的要求就是 pre[i]=suf[i+1],如果满足这个,也就是说明从前减去到a[i],从后面减去到a[i+1],只剩这两个元素,他们还是相等的,那肯定可以满足条件。
#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define pb push_back
#define me memset
const int N = 1e6 + 10;
const int mod = 1e9 + 7;
const ll INF = 1e18;
using namespace std;
typedef pair<int,int> PII;
typedef pair<ll,ll> PLL;
ll pre[N],suf[N],a[N];
int main()
{
int t;
cin>>t;
while(t--)
{
int n;
cin>>n;
for(ll i=0;i<=n+10;i++) pre[i]=suf[i]=0;
for(int i=1 ; i<=n ; i++) cin>>a[i];
for(int i=1 ; i<=n ; i++)
{
if(pre[i-1]>a[i]) pre[i]=INF/2;
else pre[i]=a[i]-pre[i-1];
}
for(int i=n ; i>=1 ; i--)
{
if(suf[i+1]>a[i]) suf[i]=INF;
else suf[i]=a[i]-suf[i+1];
}
/* for(int i=1 ; i<=n ; i++) cout<<pre[i]<<" ";
cout<<endl;
for(int i=1 ; i<=n ; i++) cout<<suf[i]<<" ";
cout<<endl;*/
bool flag=false;
for(int i=1 ; i<=n ; i++)
{
if(suf[i+1]==pre[i])
{
flag=true;
break;
}
else
{
// cout<<i<<" "<<a[i]<<" "<<a[i+1]<<" "<<pre[i-1]<<" "<<suf[i+1]<<endl;
if(a[i+1]>=pre[i-1]&&a[i]>=suf[i+2])
{
ll x=a[i+1]-pre[i-1];
ll y=a[i]-suf[i+2];
// cout<<i<<" "<<x<<" "<<y<<endl;
if(x==y)
{
flag=true;
break;
}
}
}
}
if(flag) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
// cout<<INF<<endl;
return 0;
}
总结
对性质的挖掘不到位,而且以前没有这种写题的意识,自己的代码能力还是太差。