最小异或生成树 ————codeforce 888 G

题意:给出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;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值