CF817E Choosing The Commander

CF817E Choosing The Commander

题意：
有$q$次操作，每次操作有三种类型，分别是
$1p_i$把 $p_i$ 加入集合 $S$
$2p_i$ 把 $p_i$ 从集合$S$中删除
$3p_i,l_i$ 表示查询集合中有多少个元素异或上$p_i$后小于$l_i$
集合为可重集。

思路：01Trie
一般用01Trie树来解决异或问题，我们建一颗01Trie,加入和删除就是正常Trie的操作，查询时，我们从高位开始操纵，要求异或$p_i$小于$r_i$，当$p_i$第$i$位为$0$,$l_i$第$i$位为$1$，我们就直接加上Trie树当前位为$0$的子树大小（就是以到第$i$位这些数为前缀的所有数），然后往当前位为$1$的子树走（之后有可能有小于的情况），当$p_i$第$i$位为$1$,$l_i$第$i$位为$1$，我们就直接加上Trie树当前位为$1$的子树大小，然后往当前位为$0$的子树走
细节：
这里$idx$从一开始，因为我们往子树走的时候可能没有这种情况$son[p][u]=0$,这就和根节点一样了，$cnt[p]$表示当前这包含这个结点$p$的数的个数

#include <bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for(int i = (a); i <= (b); ++i)
#define per(i, a, b) for(int i = (a); i >= (b); --i)
#define debug(x) cout << #x << " = " << x << endl;
#define ll long long
#define SZ(x) ((int)(x).size())
#define ALL(x) (x).begin(), (x).end()
const int N = 1e5 + 10, M = 2 * N;
const ll mod = 1e9 + 7;
const ll inf = 0x3f3f3f3f;

int q;
int son[30 * N][2], cnt[30 * N], idx = 1;

void add(int x, int v) {
	int p = 1;
	per(i, 30, 0) {
		cnt[p] += v;
		int u = x >> i & 1;
		if(!son[p][u]) son[p][u] = ++idx;
		p = son[p][u];
	}
	cnt[p] += v;
}

int ask(int x, int y) {
	int p = 1, res = 0;
	per(i, 30, 0) {
		int u = x >> i & 1, v = y >> i & 1;
		if(u == 0 && v == 1) res += cnt[son[p][0]];
		if(u == 1 && v == 1) res += cnt[son[p][1]];
		if(v == 1)
			p = son[p][u ^ 1];
		else
			p = son[p][u];
	}
	return res;
}

void solve() {
	cin >> q;
	rep(i, 1, q) {
		int op, p, l;
		cin >> op >> p;
		if(op == 1) {
			add(p, 1);
		} else if(op == 2) {
			add(p, -1);
		} else {
			cin >> l;
			cout << ask(p, l) << endl;
		}
	}
}

signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	// int _;cin>>_;while(_--)
	solve();
	return 0;
}