题目大意:
总共有q次询问,对于每次询问有三种不同的操作;
p = 1,增加一个数字
p = 2,去掉一个数字
p = 3,对于当前存在的数字,有x,s,问有多少 num[i]^x < s?
题解:
用字典树存所有的数字的二进制形式,sum存每种类型的个数,对于每次询问的复杂度都是log(n)。具体看代码。
代码:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <stack>
#include <bitset>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <algorithm>
using namespace std;
#define clr(a,b) memset(a,b,sizeof(a))
#define pb(a) push_back(a)
#define fir first
#define se second
#define LL long long
typedef pair<int,int> pii;
typedef pair<LL,int> pli;
typedef pair<LL,LL> pll;
const int maxn = 1e5+5;
const LL inf = 0x3f3f3f3f3f3f3f3f;
LL mod = 1e9+7;
double eps = 0.00000001;
double PI = acos(-1);
//用数组来模拟建树
struct BitTrie {
int next[maxn*31][2],fa[maxn*31],sum[maxn*31],cnt;
//next存下一个节点的下标,fa存父节点的下标,sum存当前节点的存在的所有数字个数,cnt即节点数
void init() { //初始化
cnt = 0;
clr(next,-1);
clr(sum,0);
clr(fa,-1);
}
void add(int x,int cn) {
int p = 0;
for(int i = 31;i >= 0;i--) {
int ff = (x>>i)&1;
if(next[p][ff] == -1) { //如果没有就新建节点
next[p][ff] = ++cnt;
}
fa[next[p][ff]] = p; //保存父节点
p = next[p][ff]; //找到下一个节点
}
sum[p] += cn;
//for(int i = 0;i <= 10;i++) printf("%d ",fa[i]);
//cout<<endl;
while(p != -1) { //回溯把所有子节点的个数加到父节点
int nfa = fa[p];
sum[nfa] = 0;
if(next[nfa][0] != -1) sum[nfa] += sum[next[nfa][0]];
if(next[nfa][1] != -1) sum[nfa] += sum[next[nfa][1]];
p = fa[p];
}
}
int query(int pp,int l) {
int ans = 0,p = 0;
for(int i = 31;i >= 0;i--) {
int idp = (pp>>i)&1;
int idl = (l>>i)&1;
if(idl) { //如果位置i上l的二进制值是1,那么直接把二进制值是0的加到答案中,因为二进制值是0那么所有子节点都小于l,并且往下搜二进制值是1的子节点
if(next[p][idp] != -1) ans += sum[next[p][idp]];
if(next[p][!idp] != -1) p = next[p][!idp];
else break;
}
else { //如果位置i上l的二进制值是0,那么往下搜二进制值是0的子节点
if(next[p][idp] != -1) p = next[p][idp];
else break;
}
}
return ans;
}
};
BitTrie tt;
int main() {
int q;
cin>>q;
tt.init();
while(q--) {
int k,p,l;
scanf("%d",&k);
if(k == 1) {
scanf("%d",&p);
tt.add(p,1);
}
if(k == 2) {
scanf("%d",&p);
tt.add(p,-1);
}
if(k == 3) {
scanf("%d%d",&p,&l);
printf("%d\n",tt.query(p,l));
}
}
}
入门了字典树吧,感觉字典树也不是很难。
come on!