首页 > 代码库 > BZOJ3689 异或之
BZOJ3689 异或之
题目大意:给定n个非负整数A[1], A[2], ……, A[n]。
对于每对(i, j)满足1 <= i < j <= n,得到一个新的数A[i] xor A[j],这样共有n*(n-1)/2个新的数。求这些数(不包含A[i])中前k小的数。
对于每对(i, j)满足1 <= i < j <= n,得到一个新的数A[i] xor A[j],这样共有n*(n-1)/2个新的数。求这些数(不包含A[i])中前k小的数。
注:xor对应于pascal中的“xor”,C++中的“^”。
思路:同NOI2010超级钢琴。http://blog.csdn.net/wyfcyx_forever/article/details/40400327
我们只需一开始在全局堆中放进去每个数和他之后的数异或的最小值,然后在每次出堆的的时候再放进堆中第k+1小值即可。这样重复k次。
那么问题就变成了快速求某段区间给定一个数进行xor的第k小值,我们利用可持久化trie维护size直接在树上二分即可。
时间复杂度依旧O(klogn).
Code:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; #define N 100010 int w[N]; int ch[N<<5][2], size[N<<5], ind; int root[N]; int Newadd(int Last, int bit, int x) { int q = ++ind; ch[q][0] = ch[Last][0], ch[q][1] = ch[Last][1]; size[q] = size[Last] + 1; if (bit < 0) return q; if ((x >> bit) & 1) ch[q][1] = Newadd(ch[Last][1], bit - 1, x); else ch[q][0] = Newadd(ch[Last][0], bit - 1, x); return q; } int getkth(int Left, int Right, int x, int k) { int res = 0, better; bool d; for(int dep = 30; dep >= 0; --dep) { d = (x >> dep) & 1; better = size[ch[Right][d]] - size[ch[Left][d]]; if (better >= k) Left = ch[Left][d], Right = ch[Right][d]; else Left = ch[Left][d ^ 1], Right = ch[Right][d ^ 1], res += (1<<dep), k -= better; } return res; } struct st { int end, kth, val; st(int _end = 0, int _kth = 0, int _val = 0):end(_end),kth(_kth),val(_val){} bool operator < (const st &B) const { return val < B.val; } }; #define K 250010 struct Heap { st a[K]; int top; Heap():top(0){} inline void up(int x) { for(; x != 1; x >>= 1) if (a[x]<a[x>>1]) swap(a[x],a[x>>1]); else break; } inline void down(int x) { int son; for(; (x<<1)<=top; ) { son=(((x<<1)==top)||(a[x<<1]<a[(x<<1)|1]))?(x<<1):((x<<1)|1); if (a[son]<a[x]) swap(a[son],a[x]),x=son; else break; } } inline void push(const st &x) { a[++top] = x; up(top); } st Min() { return a[1]; } inline void pop() { a[1] = a[top--]; down(1); } }H; int main() { int n, k; scanf("%d%d", &n, &k); register int i, j; for(i = 1; i <= n; ++i) { scanf("%d", &w[i]); root[i] = Newadd(root[i - 1], 30, w[i]); } for(i = 2; i <= n; ++i) H.push(st(i, 1, getkth(root[0], root[i - 1], w[i], 1))); for(i = 1; i <= k; ++i) { st tmp = H.Min(); H.pop(); if (i != 1) putchar(' '); printf("%d", tmp.val); if (tmp.kth != tmp.end - 1) H.push(st(tmp.end, tmp.kth + 1, getkth(root[0], root[tmp.end - 1], w[tmp.end], tmp.kth + 1))); } return 0; }
BZOJ3689 异或之
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。