思路:(本人也不知道算不算是,有点像两个队列依次归并)
用和归并排序一样的思路;将两个序列,按大小依次取出堆顶的部分,直到选出要求的前k个值
简单十六 异或之
H[FJ2014]异或之
时间限制 : - MS 空间限制 : - KB 评测说明 : 1s,256m
问题描述
给定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小的数。
注:xor对应于pascal中的“xor”,C++中的“^”。
solution:
- 先用trie树存储所有的数字,方便去匹配每个数对应的第k小的数
//查询操作,找第k小
int find(int x,int rk)
{
int ans = 0, u = 0;
for(int i=31;i>=0;i--)
{
int ch = (x>>i)&1;
if(cnt[go[u][ch]]>=rk)u = go[u][ch],ans <<= 1;
else rk -= cnt[go[u][ch]],u = go[u][ch^1],ans = (ans<<1|1);
}
return ans;
}
- 将所有的数匹配的第一小先放进堆里,然后被依次取出,并逐渐加入原本的数字还能够匹配出来的最小的数;(由于匹配出来的数是一对数字对亦或出来的,所以我们可以认为的选取一半用来输出)
for(int i=1;;i++)
{
node u = Q.top();Q.pop();
if(i & 1){printf("%d ",u.v);++tot;}
if(tot==k)break;
if(u.k<n)Q.push((node){find(u.a,u.k+1),u.k+1,u.a});
}
virtual judge 3 P题
题意简化:
你有k个数列,每个数列中有k个数,从每一个数列中取一个数出来和为sum,输出前k小的sum;
solution: 显然这个题需要的对于每个队列要去选择第k小不太现实,所以考虑两两归并,将连个队列归并后取出前k小的组合出来的数组成一个新的队列,那么再加一个新的数列进来,所以这样不仅可以满足前k小的数,并且可以保证每一个数都含有所有的队列中的数.;