Codeforces842 D. Vitya and Strange Lesson(01字典树,递归写法)

题意:

给定长度为n的序列a,
m次操作,每次操作给定一个整数x,将序列中的所有树都异或上x,
问每次操作之后序列a的mex是多少。

数据范围:n,m<=3e5,0<=a(i),x<=1e5

解法:

维护序列的mex,可以对建立权值线段树,在权值线段树上二分:
如果左子树满,说明mex在右子树,
如果左子树不满,说明在左子树。

这题在01字典树上二分即可,也是左边满就找右边。
因为要维护字典树每个节点的子树大小sz[],
所以字典树的插入操作用的是递归写法,便于pushup。

code:
#include<bits/stdc++.h>
using namespace std;
const int maxm=2e5+5;
struct Trie{
    int a[maxm*30][2],tot=1;
    int sz[maxm*30];
    inline void pp(int node){
        sz[node]=sz[a[node][0]]+sz[a[node][1]];
    }
    void add(int x,int i,int last){
        if(i==-1){
            sz[last]=1;
            return ;
        }
        int v=(x>>i&1);
        if(!a[last][v])a[last][v]=++tot;
        add(x,i-1,a[last][v]);
        pp(last);
    }
    int ask(int x){
        int node=1;
        int ans=0;
        for(int i=30;i>=0;i--){
            int v=(x>>i&1);
            if(sz[a[node][v]]==(1<<i)){
                node=a[node][v^1];
                ans|=(1<<i);
            }else{
                node=a[node][v];
            }
            if(!node)return ans;
        }
        return ans;
    }
}T;
int a[maxm];
int n,m;
signed main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        int x;cin>>x;
        T.add(x,30,1);
    }
    int sum=0;
    while(m--){
        int x;cin>>x;
        sum^=x;
        int ans=T.ask(sum);
        cout<<ans<<endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值