今日头条编程第二题 - 异或 (字典树)

题目

 

 

POINT：

把所有数从高到低加入01字典树。

然后对每一个数载字典树里找。

对于数x，如果当前位数的M有1，那么对答案有贡献的肯定是和x不一样的另一边，往x^1走，这样异或后才能是1（跟上M的脚步）

如果当前位数的M是0，那么我们有机会大于M，对答案的贡献就是和x不一样的另一边的子树。然后往x走。

 

#include <stdio.h>
#include <iostream>
#include <math.h>
#include <map>
using namespace std;
#define LL long long
const int N = 1e5+55;
int sz=0;
int tree[N<<1][2];
int num[N<<1];

int a[N];

void add(int x)
{
	int now = 0;
	for(int i=20;i>=0;i--){
		int to=(x&(1<<i))>0;
		if(tree[now][to]==0){
			tree[now][to]=++sz;
		}
		now=tree[now][to];
		num[now]++;
	}
}
int n,m;
int query(int x)
{
	int ans=0;
	int now = 0;
	for(int i=20;i>=0;i--){
		int to=(x&(1<<i))>0;
		if(m&(1<<i)){
			now=tree[now][to^1];
		}else{
			ans+=num[tree[now][to^1]];
			now=tree[now][to];
		}
		if(now==0) break;

	}
	return ans;
}

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		add(a[i]);
	}
	LL ans=0;
	for(int i=1;i<=n;i++){
		ans+=query(a[i]);
	}
	printf("%lld\n",ans/2);
}