icpc2020济南K Kth Query（01trie）

题意

令 $f(a,S,K)$ 表示数组 $b$ 中第 $K$ 小的元素，其中 $\forall i\in [1,n],b_i=a_i\oplus S$（$\oplus$ 为异或操作）。
现在有数组 $a$ 和 $q$ 次询问，每次询问给出 $L,R$，求 $\underset{S=L}{\overset{R}{\min }}f(a,S,K)$。
其中，$n,q\le 10^5,0\le a_i<2^{30},0\le L\le R<2^{30}$。

分析

建出 $01trie$ 后，对于 $trie$ 树上的每个节点 $p$，可以预处理出对于任意 $S\in [0,2^{30})$，$f(a,S)$ 的第 $k$ 大值的最小值，记为 $kt{h}_{p,k}$。就是一个简单的树形 $dp$。具体来说，假设从高到低来到第 $d$ 位，我们枚举 $S$ 最高位的值 $i$，有：
$$kt{h}_{p,k}=\{\begin{array}{l}kt{h}_{c{h}_{p,i},k}(siz{e}_{c{h}_{p,i}}\ge k)\\ 2^d+kt{h}_{c{h}_{p,1-i},k-siz{e}_{c{h}_{p,i}}}(otherwise)\end{array}$$
这里需要开的空间是 $O(nlogV)$ 的，可以考虑每个叶子节点的贡献，由于有 $n$ 个叶子节点，树高为 $O(logV)$，因此总空间为 $O(nlogV)$。
现在考虑有 $L\le S\le R$ 的限制。
我们从高到低考虑 $S$ 的每一位，一开始 $L,R$ 的那一位都是一样的，于是 $S$ 的选择也是唯一的。
当第一次出现分歧时，假设当前来到第 $d$ 位，即 $L_d=0,R_d=1$。这时候如果 $S_d$ 选了 $0$，那么选择范围变成了 $[L^{\prime },2^d-1)$，相当于是没有上界；选择了 $1$，那么选择范围变成了 $[2^d,R^{\prime }]$，相当于没有下界。
然后如果来到没有上下界的情况，直接返回 $kt{h}_{p,k}$ 即可，否则就一直递归下去。
复杂度是 $O((n+q)logV)$ 的。

代码

#include <bits/stdc++.h>
#define all(x) x.begin(), x.end()
#define pii pair<int, int>
#define fi first
#define se second
using namespace std;
typedef long long LL;

void debug_out(){
    cerr << endl;
}
template<typename Head, typename... Tail>
void debug_out(Head H, Tail... T){
    cerr << " " << to_string(H);
    debug_out(T...);
}
#ifdef local
#define debug(...) cerr<<"["<<#__VA_ARGS__<<"]:",debug_out(__VA_ARGS__)
#else
#define debug(...) 55
#endif

const int N = 1e5 + 5;
int ch[N * 30][2], sz[N * 30], cnt;
vector<int> kth[N * 30];
void insert(int x){
	int p = 1;
	sz[p]++;
	for(int i = 29; i >= 0; i--){
		int c = (x >> i & 1);
		if(!ch[p][c]) ch[p][c] = ++cnt;
		p = ch[p][c];
		sz[p]++;
	}
}
void dfs(int p, int d){
	if(d == -1){
		for(int i = 1; i <= sz[p]; i++) kth[p][i] = 0;
		return;
	}
	for(int i = 0; i < 2; i++) if(ch[p][i]) dfs(ch[p][i], d - 1);
	for(int i = 0; i < 2; i++){
		if(ch[p][i]){
			for(int j = 1, tmp; j <= sz[p]; j++){
				if(sz[ch[p][i]] >= j) tmp = kth[ch[p][i]][j];
				else tmp = (1 << d) + kth[ch[p][i ^ 1]][j - sz[ch[p][i]]];
				kth[p][j] = min(kth[p][j], tmp);
			}
		}
	}
}
int solve(int d, int p, int l, int r, int k){
	if(d < 0) return 0;
	if(l == 0 && r == -1){
		return kth[p][k];
	}
	if(r == -1){
		int down = (l >> d & 1), ans = (1 << 30);
		for(int i = down, tmp; i <= 1; i++){
			if(sz[ch[p][i]] >= k) tmp = solve(d - 1, ch[p][i], (i == down)? l ^ (i << d): 0, -1, k);
			else tmp = (1 << d) + solve(d - 1, ch[p][i ^ 1], (i == down)? l ^ (i << d): 0, -1, k - sz[ch[p][i]]);
			ans = min(ans, tmp);
		}
		return ans;
	}
	if(l == 0){
		int up = (r >> d & 1), ans = (1 << 30);
		for(int i = 0, tmp; i <= up; i++){
			if(sz[ch[p][i]] >= k) tmp = solve(d - 1, ch[p][i], 0, (i == up)? (r ^ (i << d)): -1, k);
			else tmp = (1 << d) + solve(d - 1, ch[p][i ^ 1], 0, (i == up)? (r ^ (i << d)): -1, k - sz[ch[p][i]]);
			ans = min(ans, tmp);
		}
		return ans;
	}
	int x = (l >> d & 1), y = (r >> d & 1);
	assert(x <= y);
	if(x == y){
		int i = x, tmp;
		if(sz[ch[p][i]] >= k) tmp = solve(d - 1, ch[p][i], l ^ (i << d), r ^ (i << d), k);
		else tmp = (1 << d) + solve(d - 1, ch[p][i ^ 1], l ^ (i << d), r ^ (i << d), k - sz[ch[p][i]]);
		return tmp;
	}
	{
		int i = 0, tmp1, tmp2;
		if(sz[ch[p][i]] >= k) tmp1 = solve(d - 1, ch[p][i], l, -1, k);
		else tmp1 = (1 << d) + solve(d - 1, ch[p][i ^ 1], l, -1, k - sz[ch[p][i]]);
		i = 1;
		if(sz[ch[p][i]] >= k) tmp2 = solve(d - 1, ch[p][i], 0, r ^ (i << d), k);
		else tmp2 = (1 << d) + solve(d - 1, ch[p][i ^ 1], 0, r ^ (i << d), k - sz[ch[p][i]]);
		return min(tmp1, tmp2);
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int n, q;
	cin >> n >> q;
	cnt = 1;
	for(int i = 0, x; i < n; i++) cin >> x, insert(x);
	for(int i = 1; i <= cnt; i++) kth[i].resize(sz[i] + 1), fill(all(kth[i]), (1 << 30));
	dfs(1, 29);
	//debug(kth[1][1], kth[1][2], kth[1][3]);
	while(q--){
		int l, r, k;
		cin >> l >> r >> k;
		cout << solve(29, 1, l, r, k) << '\n';
	}
	return 0;
}