D. Cleaning

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

总结
对性质的挖掘不到位,而且以前没有这种写题的意识,自己的代码能力还是太差。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值