codeforces1438 E.Yurii Can Do Everything

E.Yurii Can Do Everything

官方题解
按照题解的意思,由于满足此条件的数量不多,因此可以挖掘某些信息进行暴力。

考虑一个good subarray: [ a l , a l + 1 , … , a r − 1 , a r ] [a_l,a_{l+1},\dots,a_{r-1},a_r] [al,al+1,,ar1,ar]
按照题目意思满足 a l ⊕ a r = ∑ i = l + 1 r − 1 a i a_l\oplus a_r=\sum_{i=l+1}^{r-1}a_i alar=i=l+1r1ai,如果设 a l , a r a_l,a_r al,ar二进制表示中最高位是分别为 k l , k r k_l,k_r kl,kr,那么一定有 2 m a x ( k l , k r ) + 1 > ∑ i = l + 1 r − 1 a i 2^{max(k_l,k_r)+1}>\sum_{i=l+1}^{r-1}a_i 2max(kl,kr)+1>i=l+1r1ai,由此我们正序逆序分别处理一下,在不满足上述条件的情况下提前跳出循环,注意可能重复计算

由于我们逆序处理,只需要每次把和与 2 k l + 1 2^{k_l+1} 2kl+1比较即可,因为逆序后如果最开始右端点满足题意同样会枚举到,如果原来的 2 k l + 1 2^{k_l+1} 2kl+1 2 k r + 1 2^{k_r+1} 2kr+1都大于和,那么可能会重复计算

#define IO ios::sync_with_stdio(false);cin.tie();cout.tie(0)
#pragma GCC optimize(2)
#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<random>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<unordered_map>
#include<unordered_set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int N=200010;
map<pii,int>mp;
int n,a[N],res;
void solve(bool op)
{
    for(int i=1;i+2<=n;i++)
    {
        ll s=a[i+1];
        int k=0;
        for(int j=0;j<=29;j++)
            if(a[i]>>j&1) k=j;
        if(!op)
        {
            for(int j=i+2;j<=n;j++)
            {
                if(s==(a[i]^a[j])) res++,mp[{n-j+1,n-i+1}]=1;
                s+=a[j];
                if(s>=(1<<k+1)) break;
            }
        }
        else
        {
            for(int j=i+2;j<=n;j++)
            {
                if(s==(a[i]^a[j])&&!mp.count({i,j})) res++;
                s+=a[j];
                if(s>=(1<<k+1)) break;
            }
        }
    }
}
int main()
{
    IO;
    int T=1;
    //cin>>T;
    while(T--)
    {
        cin>>n;
        for(int i=1;i<=n;i++) cin>>a[i];
        solve(0);
        reverse(a+1,a+1+n);
        solve(1);
        cout<<res<<'\n';
    }
    return 0;
}

总结:如果自己感觉满足题意的个数不多,可以考虑挖掘某些信息进行暴力,提前跳出循环从而满足时间复杂度

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值