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;
}