G. Xor-MST
time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
You are given a complete undirected graph with n vertices. A number ai is assigned to each vertex, and the weight of an edge between vertices i and j is equal to ai xor aj.
Calculate the weight of the minimum spanning tree in this graph.
Input
The first line contains n (1 ≤ n ≤ 200000) — the number of vertices in the graph.
The second line contains n integers a1, a2, ..., an (0 ≤ ai < 230) — the numbers assigned to the vertices.
Output
Print one number — the weight of the minimum spanning tree in the graph.
Examples
input
Copy
5 1 2 3 4 5
output
8
input
Copy
4 1 2 3 4
output
8
题意:给你n个点的点权,连接两个点的代价为ai^aj,求最小生成树。
我们可以按位进行计算,把0和1分开递归,如果只有最低位不一样,那么一定要连边。
在合并的时候,我们要把1中一个数和0中一个数能够得到的异或最小值合并,我们用01trie树进行查询即可,每层暴力建树,下一层就要把树销毁。
这样的贪心是对的,因为二进制一个数字比地位所有数字加起来还要大。
我们每次都把能用到的最小代价用到了,所以不会错。
代码:
#include<bits/stdc++.h>
using namespace std;
int nex[7000005][2],tot,a[200005],n;
long long ans=0;
void ins(int x)
{
int now=0;
for(int i=29;i>=0;i--)
{
int t=(x>>i)&1;
if(nex[now][t]==0) nex[now][t]=++tot;
now=nex[now][t];
}
}
int ask(int x)
{
int res=0,now=0;
for(int i=29;i>=0;i--)
{
int t=(x>>i)&1;
if(nex[now][t]) now=nex[now][t];
else now=nex[now][t^1],res|=(1<<i);
}
return res;
}
void dfs(int l,int r,int dep)
{
if(dep==-1||l>=r) return;
int mid=l-1;
while(mid<r&&(a[mid+1]>>dep&1)==0) mid++;
dfs(l,mid,dep-1),dfs(mid+1,r,dep-1);
if(mid==l-1||mid==r) return;
for(int i=l;i<=mid;i++) ins(a[i]);
int res=2147483647;
for(int i=mid+1;i<=r;i++) res=min(res,ask(a[i]));
ans+=res;
for(int i=0;i<=tot;i++) nex[i][0]=nex[i][1]=0;
tot=0;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
sort(a+1,a+n+1);
dfs(1,n,29);
cout<<ans<<endl;
}