BZOJ 4103 [Thu Summer Camp 2015]异或运算 可持久化Trie

对矩阵求第k大xor值。

注意到数据范围长宽严重失调（误

所以对于每列维护可持久化Trie。

然后查询就由BZOJ 3261推广一下就好啦。

#include <cstdio>
#include <cstring>
#include <algorithm>
const int N = 1001, M = 300001;
using namespace std;
struct Trie {
Trie* c[2]; int size;
Trie() { c[0] = c[1] = NULL; size = 0; }
Trie(Trie *l, Trie *r, int sz) { c[0] = l, c[1] = r, size = sz; }
void* operator new(size_t) {
static Trie pool[M * 60], *pointer = pool;
return pointer++;
}
static Trie* put(Trie *p, int x, int digit) {
if (!digit)
return new Trie(p->c[0], p->c[1], p->size + 1);
if (x & digit)
return new Trie(p->c[0], put(p->c[1], x, digit >> 1), p->size + 1);
else
return new Trie(put(p->c[0], x, digit >> 1), p->c[1], p->size + 1);
}
static int get(Trie *p1[], Trie *p2[], int *x, int len, int digit, int k) {
int i, sz = 0;
if (!digit)
return 0;
#define rson (bool(~x[i] & digit))
#define son (bool( x[i] & digit))
for (i = 1; i <= len; i++)
sz += p1[i]->c[rson]->size,
sz -= p2[i]->c[rson]->size;
if (k <= sz) {
for (i = 1; i <= len; i++)
p1[i] = p1[i]->c[rson],
p2[i] = p2[i]->c[rson];
return digit | get(p1, p2, x, len, digit >> 1, k);
} else {
for (i = 1; i <= len; i++)
p1[i] = p1[i]->c[ son],
p2[i] = p2[i]->c[ son];
return get(p1, p2, x, len, digit >> 1, k - sz);
}
}
};
int main() {
static int x
, y[M], a
;
static Trie *p1
, *p2
, *trie[M];
int i, dep = 1 << 30, n, m, q, l1, l2, r1, r2, k;
scanf("%d%d", &n, &m);
for (i = 1; i <= n; i++) scanf("%d", &x[i]);
for (i = 1; i <= m; i++) scanf("%d", &y[i]);
trie[0] = new Trie();
trie[0]->c[0] = trie[0]->c[1] = trie[0];
for (i = 1; i <= m; i++)
trie[i] = Trie::put(trie[i - 1], y[i], dep);
scanf("%d", &q);
while (q--) {
scanf("%d%d%d%d%d", &l1, &r1, &l2, &r2, &k);
int len = r1 - l1 + 1;
fill(p1 + 1, p1 + len + 1, trie[r2]);
fill(p2 + 1, p2 + len + 1, trie[l2 - 1]);
memcpy(a + 1, x + l1, sizeof(int) * len);
printf("%d\n", Trie::get(p1, p2, a, len, dep, k));
}
return 0;
}

4103: [Thu Summer Camp 2015]异或运算

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 171  Solved: 96

[Submit][Status][Discuss]

Description

给定长度为n的数列X={x1,x2,...,xn}和长度为m的数列Y={y1,y2,...,ym}，令矩阵A中第i行第j列的值Aij=xi xor  yj，每次询问给定矩形区域i∈[u,d],j∈[l,r]，找出第k大的Aij。

Input

第一行包含两个正整数n,m，分别表示两个数列的长度
第二行包含n个非负整数xi
第三行包含m个非负整数yj
第四行包含一个正整数p，表示询问次数
随后p行，每行均包含5个正整数，用来描述一次询问，每行包含五个正整数u,d,l,r,k，含义如题意所述。

Output

共p行，每行包含一个非负整数，表示此次询问的答案。

Sample Input

3 3

1 2 4

7 6 5

3

1 2 1 2 2

1 2 1 3 4

2 3 2 3 4

Sample Output

6

5

1

HINT

 对于100%的数据，0<=Xi,Yj<2^31,

1<=u<=d<=n<=1000,

1<=l<=r<=m<=300000,

1<=k<=(d-u+1)*(r-l+1),

1<=p<=500