G. Xor-MST
题意:
给你
2
e
5
2e5
2e5 个点,每个点 的权值小于等于
1
e
9
1e9
1e9;
每两个点之间都有连边,边权是 两点权值的异或值,求最小生成树;
思路:
本来莽了一发 Boruvka,可能是写法问题 MLE了,换了一种分治的写法。
考虑将所有点权排序,找到最高位的 使得把 集合分成两个部分,两个部分的集合 分治处理,两个集合 之间的连边 可以用字典树 处理出来,就AC了
原理是基于 异或的高位贪心,不仔细讲了吧……
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int n;
struct node{
int vis[2];
}trie[1650060];
int tot=0;
int s[50];
ll ans=0;
int a[200050];
void insert(){
int now=0;
for(int i=0;i<30;i++){
if(trie[now].vis[s[i]])now=trie[now].vis[s[i]];
else {
trie[now].vis[s[i]]=++tot;
now=trie[now].vis[s[i]];
trie[now].vis[0]=trie[now].vis[1]=0;
}
}
}
int find(){
int now=0;
int res=0;
for(int i=0;i<30;i++){
if(s[i]==0){
if(trie[now].vis[0]){
now=trie[now].vis[0];
res=res*2;
}else res=res*2+1,now=trie[now].vis[1];
}else{
if(trie[now].vis[1]){
now=trie[now].vis[1];
res=res*2;
}else res=res<<1|1,now=trie[now].vis[0];
}
}
return res;
}
void solve(int l,int r,int dep){
if(dep<0||l>=r)return ;
int mid=l-1;
while(mid<r&&((a[mid+1]>>dep)&1)==0)mid++;
solve(l,mid,dep-1);
solve(mid+1,r,dep-1);
if(mid==l-1||mid==r)return;
tot=0;
trie[0].vis[0]=trie[0].vis[1]=0;
for(int i=l;i<=mid;i++){
int p=a[i];
for(int j=29;j>=0;j--){
s[j]=p&1;
p>>=1;
}
insert();
}
int tmp=2e9;
for(int i=mid+1;i<=r;i++){
int p=a[i];
for(int j=29;j>=0;j--){
s[j]=p&1;
p>>=1;
}
tmp=min(tmp,find());
}
ans+=tmp;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
sort(a+1,a+1+n);
solve(1,n,29);
printf("%lld\n",ans);
return 0;
}