3689: 异或之

题目链接

题目大意:给定n个数,求这n个数两两异或的值中的前k小

题解:经典套路……
把每个数按照二进制建立Trie,可以利用Trie树上的size域查询出一个数与其它数异或值的第k小

维护一个堆,将所有数与其它异或值的第2小加入堆(第一小是自己异或自己,不在题目要求范围内),当取出一个数异或值的第k小后,将第k+1小加入堆

一个异或值会被两个数分别取出一次,所以取出奇数次时输出,取2*k次即可

我的收获:23333

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;

const int N=100015;

int n,k;
int a[N];

struct lcr{int w,a,rk;};

bool operator <(lcr x,lcr y){return x.w>y.w;}

priority_queue<lcr> q;

struct Trie{

int tot;
int size[N*30];
int c[N*30][2];

void insert(int s){
    int x=0;
    for(int i=30;i>=0;i--){
        bool idx=s&(1<<i);
        if(!c[x][idx]) c[x][idx]=++tot;
        x=c[x][idx];size[x]++;
    }
}

int query(int s,int k){
    int x=0,temp=0;
    for(int i=30;i>=0;i--){
        bool idx=s&(1<<i);
        if(size[c[x][idx]]>=k) x=c[x][idx];
        else k-=size[c[x][idx]],temp+=1<<i,x=c[x][idx^1];
    }
    return temp;
}

}T;

void work()
{
    for(int i=1;i<=n;i++){
        lcr x;
        x.w=T.query(a[i],2);x.rk=2;x.a=a[i];
        q.push(x);
    }
    for(int i=1;i<=k*2;i++){
        lcr x=q.top();q.pop();
        if(i&1) printf("%d ",x.w);
        if(x.rk==n) continue;
        x.rk++;x.w=T.query(x.a,x.rk);
        q.push(x);
    }
}

void init()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),T.insert(a[i]); 
}

int main()
{
    init();
    work();
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值