CodeForces - 665E Beautiful Subarrays (01字典树,异或)

题意:给你一个数组,问你该数组有多少子区间异或和大于等于k
问到区间异或和首先想到前缀和,定义sum[ i ] 为1 ~ i 的异或值
对于每个r,查找有多少sum[ i ] & sum[ r ] >=k,首先想到01字典树,从高位像低位贪心。如果该位有异或值大于K的情况,如x=1,k=0,那么先加上cnt[son [ p1 ] ](因为该位已经大于了,后面不用再讨论了),然后让p = son[ p0] (让当前异或值保持在等于K),特判的一点就是,如果最后能走到0位( = k的情况),需要加上。

写的时候犯的蠢
(1)还是字典树大小的问题,考虑每个插入的x,会产生31个结点,所以字典树最多MX*31个结点
(2)数的异或运算是^,&是与运算(我好蠢5555)

const int MX = 1e6 + 10;
int n,k,a[MX];
ll sum[MX];
int tot,son[MX*33][2],cnt[MX*33];
void add(ll x)
{
    int p=0;
    for(int k=30;k>=0;--k)
    {
        int tmp=((x&(1ll<<k))==0?0:1);
        if(son[p][tmp]==0) son[p][tmp]=++tot;
        p=son[p][tmp]; 
        ++cnt[p];
    }
}
ll check(int x,int k)
{
    int p=0;
    ll ans=0;
    for(int i=30;i>=0;--i) 
    {
        if((x&(1ll<<i))&&(k&(1ll<<i)))//1 1
            p=son[p][0];
        else if((x&(1ll<<i))&&!(k&(1ll<<i)))//1 0 
            ans+=cnt[son[p][0]],p=son[p][1];
        else if(!(x&(1ll<<i))&&(k&(1ll<<i)))//0 1
            p=son[p][1];
        else if(!(x&(1ll<<i))&&!(k&(1ll<<i)))//0 0
            ans+=cnt[son[p][1]],p=son[p][0];
        if(p==0) break;
        if(i==0) ans+=cnt[p]; 
    }
    return ans;
}
signed main()
{
    cin>>n>>k;
    ll ans=0;
    add(0);
    for(int i=1;i<=n;++i)
    {
        cin>>a[i];
        sum[i]=sum[i-1]^a[i];
        ans+=check(sum[i],k);
        add(sum[i]);
    }
    cout<<ans<<endl;
    return  0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值