Codeforces817 E. Choosing The Commander（01字典树）

最新推荐文章于 2022-04-28 17:15:56 发布

live4m

最新推荐文章于 2022-04-28 17:15:56 发布

阅读量177

点赞数

本文链接：https://blog.csdn.net/weixin_44178736/article/details/117710814

版权

这篇博客介绍了一个使用01字典树解决动态集合操作的问题，包括添加、删除元素以及询问两个数的异或值小于某个值的元素数量。代码中详细展示了如何实现这些操作，并在主函数中进行测试。该解法主要涉及数据结构和算法的应用。

摘要由CSDN通过智能技术生成

题意：

在这里插入图片描述

解法：

容易想到01字典树,

操作1和操作2是字典树的基本操作,
对于操作3,如果在树上走x^p=l的前缀,
对于某一位,如果走另一条路径会使得x^p<l,那么加上另一条路径的size即可.

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=1e5+5;
struct Trie{
    int a[maxm*32][2],tot=1;
    int sz[maxm*32];
    void add(int x){
        int node=1;
        for(int i=30;i>=0;i--){
            int v=(x>>i&1);
            if(!a[node][v])a[node][v]=++tot;
            node=a[node][v];
            sz[node]++;
        }
    }
    void del(int x){
        int node=1;
        for(int i=30;i>=0;i--){
            int v=(x>>i&1);
            if(!a[node][v])a[node][v]=++tot;
            node=a[node][v];
            sz[node]--;
        }
    }
    int ask(int x,int y){
        int node=1;
        int ans=0;
        for(int i=30;i>=0;i--){
            int v1=(x>>i&1),v2=(y>>i&1);
            int v=(v1^v2);//走v的话,是等于v2
            if((v1^(v^1))<v2){//如果走v^1会更小,则加上贡献
                ans+=sz[a[node][v^1]];
            }
            node=a[node][v];
            if(!node)return ans;
        }
        return ans;
    }
}T;
void solve(){
    int q;cin>>q;
    while(q--){
        int op;cin>>op;
        if(op==1){
            int x;cin>>x;T.add(x);
        }else if(op==2){
            int x;cin>>x;T.del(x);
        }else if(op==3){
            int x,y;cin>>x>>y;
            int ans=T.ask(x,y);
            cout<<ans<<endl;
        }
    }
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    solve();
    return 0;
}