模拟赛 生成树

题目描述

给定 $n$ 个点，第 $i$ 个点有一个非负点权 $a_i$。
这 $n$ 个点连成了一张无向完全图，其中第 $i$ 个点和第 $j$ 个点连的边长度为 $a_i$ 按位异或 $a_j$。现在要求出这张无向完全图的最小生成树。
$1⩽n⩽200000,0⩽a_i⩽1073741823$
时间限制2s，空间限制512MB。

题解

位运算问题常用贪心。将 $a$ 排序并分为高位为$0$和高位为$1$两类。那么整个图的最小生成树则为高位为$0$的数构成的最小生成树并上高位为$1$的最小生成树再加上再两类中个选一个数的最小边权。高位相同的答案可以递归，高位不同时将高位为$0$的数加入01字典树中，再将高位为$1$的数挨个在字典树中跑，从高位选择到低位，找出最小边权。
时间复杂度 $O(nlo{g}_2^{2}ma{x}_a)$，空间复杂度 $O(nlo{g}_{2}ma{x}_a)$。

#include<stdio.h>
#include<algorithm>
using namespace std;
#define R register int
#define L long long
#define I inline
int a[200001],ct=1;
struct TrieNode{
	int Ls,Rs;
}t[6000000];
I int GetNode(){
	ct++;
	return ct;
}
I void Insert(int x){
	int p=1;
	for(R i=29;i!=-1;i--){
		if((x>>i&1)==1){
			if(t[p].Rs==0){
				t[p].Rs=GetNode();
			}
			p=t[p].Rs;
		}else{
			if(t[p].Ls==0){
				t[p].Ls=GetNode();
			}
			p=t[p].Ls;
		}
	}
}
I int GetMin(int p,int x,int k){
	int res=0;
	for(R i=k;i!=-1;i--){
		if((x>>i&1)==1){
			if(t[p].Rs==0){
				res|=1<<i;
				p=t[p].Ls;
			}else{
				p=t[p].Rs;
			}
		}else{
			if(t[p].Ls==0){
				res|=1<<i;
				p=t[p].Rs;
			}else{
				p=t[p].Ls;
			}
		}
	}
	return res;
}
I void Solve(int p,int l,int r,int k,int pre,L&ans){
	if(k!=-1){
		int mid=lower_bound(a+l,a+r+1,pre|1<<k)-a;
		if(t[p].Ls!=0){
			Solve(t[p].Ls,l,mid-1,k-1,pre,ans);
		}
		if(t[p].Rs!=0){
			Solve(t[p].Rs,mid,r,k-1,pre|1<<k,ans);
		}
		if(t[p].Ls!=0&&t[p].Rs!=0){
			int cur=2e9;
			for(R i=mid;i<=r;i++){
				int tem=GetMin(t[p].Ls,a[i],k-1);
				if(tem<cur){
					cur=tem;
				}
			}
			ans+=cur|1<<k;
		}
	}
}
int main(){
	int n;
	scanf("%d",&n);
	for(R i=1;i<=n;i++){
		scanf("%d",a+i);
		Insert(a[i]);
	}
	sort(a+1,a+1+n);
	L ans=0;
	Solve(1,1,n,29,0,ans);
	printf("%lld",ans);
	return 0;
}