CodeChef XRQRS - Xor Queries

CodeChef XRQRS - Vjudge

题目简述
给出一个一维数组和若干个不同类型的询问：
• 0 x ：在数组的最后添加 x
• 1 L R x ：在区间 [L,R] 中找到 y，使 x xor y 最大
• 2 k ：删除数组的最后 k 个数
• 3 L R x ：统计区间 [L,R] 中有多少个数 y 小于或等于 x
• 4 L R k ：找出区间 [L,R] 中从小到大的第 k 个数

题解

将每一个数按照其二进制插入 TRIE 中，维护前缀的 TRIE 树总和。

对于每一个询问，在两棵TRIE 上由上至下的走就可以了。

#include<iostream>
#include<cstdio>
#include<cstring>

#define ll long long
using namespace std;
const int N = 500000 + 10;
int rt
;
int siz
,son
[2];
int tot;
int m;
int n;
void change(int &x, int xx, int d, int p){
x = ++tot;
memcpy(son[x], son[xx], sizeof(son[x]));
siz[x] = siz[xx] + 1;
if(d < 0)
return ;
if((p >> d-1) & 1)
change(son[x][1], son[xx][1], d-1, p);
else
change(son[x][0], son[xx][0], d-1, p);
}
int query1(int x, int xx, int d, int y){
if(d < 0)
return 0;
bool z = (y >> d-1) & 1 ^ 1;
if(siz[son[xx][z]] - siz[son[x][z]])
return (1 << d-1) + query1(son[x][z], son[xx][z], d-1, y);
else
return query1(son[x][z^1], son[xx][z^1], d-1, y);
}
int query2(int x, int xx, int d, int now, int y){
if(d<0)
return 0;
if(y >= now + (1 << d-1))
return siz[son[xx][0]] - siz[son[x][0]] + query2(son[x][1], son[xx][1], d-1, now+(1 << d-1), y);
else
return query2(son[x][0], son[xx][0], d-1, now, y);
}
int query3(int x,int xx,int d,int k){
if(d < 0)
return 0;
int t = siz[son[xx][0]] - siz[son[x][0]];
if(k <= t)
return query3(son[x][0], son[xx][0], d-1, k);
else
return (1 << d-1) + query3(son[x][1], son[xx][1], d-1, k-t);
}
int main(){
int o, x, y, z;
scanf("%d", &m);
while(m--){
scanf("%d", &o);
if(o == 0){
scanf("%d", &x);
change(rt[++n], rt
, 19 ,x);
}
if(o == 1){
scanf("%d%d%d",&x,&y,&z);
printf("%d\n",z^query1(rt[x-1],rt[y],19,z));
}
if(o == 2){
scanf("%d", &x);
n -= x;
tot = rt[n+1] - 1;
}
if(o == 3){
scanf("%d%d%d", &x, &y, &z);
printf("%d\n", query2(rt[x-1], rt[y], 19, 0, z));
}
if(o == 4){
scanf("%d%d%d", &x,&y,&z);
printf("%d\n", query3(rt[x-1], rt[y], 19, z));
}
}
return 0;
}