【BZOJ 3689】异或之 trie+堆

以前做过类似的题目，不同的是把异或改成加法。。。。。。

还以为异或和加法没有太大的区别，把初始的几个数字放进优先队列里面，一个一个的取出来，得到次大值，再放进去，一直重复知道k个，至于找一个数和其他树异或的k大值可以用trie树轻松解决。

原因嘛，很显然对于每一个数经过我们构造出来的数字是单调递增的，然后放进堆里面每次保证取出堆里面的数字也是单调递增的（有点类似于dijkstra的样子，当时是自己yy的啦）

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#define maxn 3000211
using namespace std;
int n,k,ch[maxn][2],size[maxn],tot,a[maxn];
struct node{
	int v,k,a;
	bool operator<(const node& y)const{return v>y.v;}
};
priority_queue<node>q;
void insert(int x){
	int rt=0;
	for(int c,i=31;i>=0;i--){
		c=(x>>i)&1;
		if(!ch[rt][c])ch[rt][c]=++tot;
		rt=ch[rt][c];size[rt]++;
	}
}
int query(int x,int k){
	int ans=0,rt=0;
	for(int c,i=31;i>=0;i--){
		c=(x>>i)&1;
		if(k>size[ch[rt][c]]){
			k-=size[ch[rt][c]];
			ans=(ans<<1)|1,rt=ch[rt][!c];
		}else{
			ans<<=1,rt=ch[rt][c];
		}
	}
	return ans;
}
int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)scanf("%d",a+i),insert(a[i]);
	for(int i=1;i<=n;i++)q.push((node){query(a[i],2),2,a[i]});
	for(int cnt=0,i=1;;i++){
		node u=q.top();q.pop();
		if(i&1)printf("%d ",u.v),cnt++;
		if(cnt==k)break;if(u.k==n)continue;
		q.push((node){query(u.a,u.k+1),u.k+1,u.a});	}
	return 0;
}