题意
令
f
(
a
,
S
,
K
)
f(a,S,K)
f(a,S,K) 表示数组
b
b
b 中第
K
K
K 小的元素,其中
∀
i
∈
[
1
,
n
]
,
b
i
=
a
i
⊕
S
\forall i \in [1,n], b_i=a_i \oplus S
∀i∈[1,n],bi=ai⊕S(
⊕
\oplus
⊕ 为异或操作)。
现在有数组
a
a
a 和
q
q
q 次询问,每次询问给出
L
,
R
L,R
L,R,求
min
S
=
L
R
f
(
a
,
S
,
K
)
\min\limits_{S=L}^{R}f(a,S,K)
S=LminRf(a,S,K)。
其中,
n
,
q
≤
1
0
5
,
0
≤
a
i
<
2
30
,
0
≤
L
≤
R
<
2
30
n,q\le 10^5,0\le a_i<2^{30},0\le L\le R<2^{30}
n,q≤105,0≤ai<230,0≤L≤R<230。
分析
建出
01
t
r
i
e
01trie
01trie 后,对于
t
r
i
e
trie
trie 树上的每个节点
p
p
p,可以预处理出对于任意
S
∈
[
0
,
2
30
)
S\in[0,2^{30})
S∈[0,230),
f
(
a
,
S
)
f(a,S)
f(a,S) 的第
k
k
k 大值的最小值,记为
k
t
h
p
,
k
kth_{p,k}
kthp,k。就是一个简单的树形
d
p
dp
dp。具体来说,假设从高到低来到第
d
d
d 位,我们枚举
S
S
S 最高位的值
i
i
i,有:
k
t
h
p
,
k
=
{
k
t
h
c
h
p
,
i
,
k
(
s
i
z
e
c
h
p
,
i
≥
k
)
2
d
+
k
t
h
c
h
p
,
1
−
i
,
k
−
s
i
z
e
c
h
p
,
i
(
o
t
h
e
r
w
i
s
e
)
kth_{p,k}=\begin{cases}kth_{ch_{p,i},k}\ \ (size_{ch_{p,i}}\ge k)\cr 2^d+kth_{ch_{p,1-i},k-size_{ch_{p,i}}}\ (otherwise)\end{cases}
kthp,k={kthchp,i,k (sizechp,i≥k)2d+kthchp,1−i,k−sizechp,i (otherwise)
这里需要开的空间是
O
(
n
l
o
g
V
)
O(nlogV)
O(nlogV) 的,可以考虑每个叶子节点的贡献,由于有
n
n
n 个叶子节点,树高为
O
(
l
o
g
V
)
O(logV)
O(logV),因此总空间为
O
(
n
l
o
g
V
)
O(nlogV)
O(nlogV)。
现在考虑有
L
≤
S
≤
R
L\le S\le R
L≤S≤R 的限制。
我们从高到低考虑
S
S
S 的每一位,一开始
L
,
R
L,R
L,R 的那一位都是一样的,于是
S
S
S 的选择也是唯一的。
当第一次出现分歧时,假设当前来到第
d
d
d 位,即
L
d
=
0
,
R
d
=
1
L_d=0,R_d=1
Ld=0,Rd=1。这时候如果
S
d
S_d
Sd 选了
0
0
0,那么选择范围变成了
[
L
′
,
2
d
−
1
)
[L',2^d-1)
[L′,2d−1),相当于是没有上界;选择了
1
1
1,那么选择范围变成了
[
2
d
,
R
′
]
[2^d,R']
[2d,R′],相当于没有下界。
然后如果来到没有上下界的情况,直接返回
k
t
h
p
,
k
kth_{p,k}
kthp,k 即可,否则就一直递归下去。
复杂度是
O
(
(
n
+
q
)
l
o
g
V
)
O((n+q)logV)
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;
}