题意:给出n个点的,求最小异或生成树:连接 x,y,则这条边的权值为a[x] ^a[y]
思路:
根据贪心的思路如果一个点要与另一个点相连,他与 和它同一个lca的点相连是最优的,因为此时上面的部分都异或掉了所以只需要将每个点的权值转换成二进制,从高位往低位依次插入01字典树.
建立字典树之后一个节点下的最小异或生成树 = 左子树 + 右子树 + 左右合并的最小值。
dfs遍历该字典树的每个结点,如果某个结点有两个子节点,这两个子节点的子树分别会构成两个连通块,要在这两个连通块之间各选一个点连边并使它们的异或值最小,通过递归处理这个问题.
find函数中先考虑能不能往相同的子节点(都为0或者1)因为相同两条边异或为0,返回值取这两种走法的最小值.如果不能,返回值要加上此时的二进制位对应的值,还要加上这两种走法继续向下的最小值.
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+7;
const int mod = 998244353;
typedef long long ll;
int inf=0x3f3f3f3f;
ll n,m,k;
int cnt;
ll a[maxn];
int l[maxn],r[maxn];
int tr[maxn][2];
void Insert(ll x,int id)
{
int p=0;
for(int i=32;i>=0;i--){
int u=x>>i&1;
if(!tr[p][u]) tr[p][u] = ++cnt;
p=tr[p][u];
if(!l[p]) l[p] = id;
r[p] = id;
}
}
ll fun(int p,int pos,ll x)
{
ll res=0;
for(int i=pos;i>=0;i--)
{
int u=x>>i&1;
if(tr[p][u]) p = tr[p][u];
else{
p = tr[p][!u];
res += (1<<i);
}
}
return res;
}
ll find(int p,int pos)
{
if(tr[p][0] && tr[p][1])
{
int x = tr[p][0],y = tr[p][1];
ll mins=1e17;
for(int i=l[x];i<=r[x];i++)
mins=min(mins,fun(y,pos-1,a[i])+(1<<pos));
return mins+find(tr[p][0],pos-1)+find(tr[p][1],pos-1);
}
else if(tr[p][0]) return find(tr[p][0],pos-1);
else if(tr[p][1]) return find(tr[p][1],pos-1);
return 0;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
sort(a+1,a+1+n);
for(int i=1;i<=n;i++)
Insert(a[i],i);
printf("%lld\n",find(0,32));
return 0;
}