CF888G XOR-MST 最小异或生成树

CF888G XOR-MST

链接

CF888G

思路

trie上贪心，先左右两边连边，再用一条边的代价连起左右两颗树。因为内部的边一定比跨两棵树的边权笑，显然是对的。
代码自己瞎yy的。启发式合并

代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int _=2e5+7;
int n,a[_],ch[_*30][2],siz[_*30],cnt,col,rt[_*30];
ll ans;
std::vector<int> dsr[_*30];
void insert(int x) {
    int p=0;
    for(int i=29;i>=0;--i) {
        bool k=x&(1LL<<i);
        siz[p]++;
        if(!ch[p][k]) ch[p][k]=++cnt;
        p=ch[p][k];
    } siz[p]++;
    if(!rt[p]) rt[p]=++col;
    dsr[rt[p]].push_back(x);
}
int query(int rt,int init,int x) {
    int p=rt,ans=0;
    for(int i=init;i>=0;--i) {
        bool k=x&(1LL<<i);
        if(ch[p][k]) p=ch[p][k];
        else p=ch[p][!k],ans|=1<<i;
    }
    return ans;
}
void dfs(int p,int dep) {
    if(!ch[p][0]&&!ch[p][1]) return;
    if(!ch[p][0]||!ch[p][1]) {
        dfs(ch[p][0]|ch[p][1],dep-1);
        rt[p]=rt[ch[p][0]|ch[p][1]];
    } else {
        int k=siz[ch[p][0]]<siz[ch[p][1]];
        dfs(ch[p][0],dep-1),dfs(ch[p][1],dep-1);
        rt[p]=rt[ch[p][k]];
        int tmp=0x3f3f3f3f;
        int nb=ch[p][!k];
        ch[p][!k]=0;
        for(auto x:dsr[rt[nb]]) {
            tmp=min(tmp,query(p,dep,x));
            dsr[rt[p]].push_back(x);
        }
        ans+=tmp;
        ch[p][!k]=nb;
    }
}
int main() {
    // freopen("a.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;++i) scanf("%d",&a[i]),insert(a[i]);
    dfs(0,29);
    cout<<ans<<"\n";
    return 0;
}

转载于:https://www.cnblogs.com/dsrdsr/p/11405990.html