[Educational Codeforces Round 32] - G Xor-MST

洛谷传送门

Codeforces传送门

题意翻译

已知一个$n$个节点的无向完全图，第$i$个节点的权值为$a_i$,$i$与$j$的边的权值是$a_i⨁a_j$,求该图的MST的权值

解题分析

贪心地想， 从高位往低位考虑， 我们一定是先考虑在这一位都为$1$或都为$0$的数之间连上边， 形成大的连通块， 这样能保证最后答案一定最小。

然后就有了一个类似CDQ的做法： 每次把当前位为1的搞在一起， 为0的搞在一起， 递归下去处理， 然后再考虑$01$之间的连边。

然后发现这么搞之后每次$01$块之间都只需要连一条边即可满足， CDQ也只是一个空架子， 直接上$Trie$树， 枚举左右儿子较小的子树中的所有元素， 去另一个子树中找到最优解就好了。

总复杂度$O(Nlogmax(a_i)logN)$。

代码如下：

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <cstring>
#include <climits>
#include <vector>
#include <algorithm>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define MX 205000
#define ls tree[now].son[0]
#define rs tree[now].son[1]
#define ll long long
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
template <class T> IN T max(T a, T b) {return a > b ? a : b;}
template <class T> IN T min(T a, T b) {return a < b ? a : b;}
int n, cnt,root;
struct Node
{
	int son[2];
	std::vector <int> ele;
} tree[MX * 30];
int val[MX];
ll ans;
void insert(int &now, R int val, R int dgt)
{
	if (!now) now = ++cnt;
	tree[now].ele.push_back(val);
	if (dgt < 0) return;
	int bit = (val >> dgt) & 1;
	if (!bit) insert(ls, val, dgt - 1);
	else insert(rs, val, dgt - 1);
}
int calc(R int now, R int val, R int dgt)
{
	if (dgt < 0) return 0;
	int bit = (val >> dgt) & 1;
	if (tree[now].son[bit]) return calc(tree[now].son[bit], val, dgt - 1);
	else return calc(tree[now].son[bit ^ 1], val, dgt - 1) + (1 << dgt);
}
void DFS(R int now, R int dgt)
{
	if (dgt < 0) return;
	if (ls) DFS(ls, dgt - 1);
	if (rs) DFS(rs, dgt - 1);
	if (ls && rs)
	{
		int best = INT_MAX;
		if (tree[ls].ele.size() < tree[rs].ele.size())
		for (R int i = tree[ls].ele.size() - 1; ~i; --i)
		best = min(best, calc(rs, tree[ls].ele[i], dgt - 1));
		else
		for (R int i = tree[rs].ele.size() - 1; ~i; --i)
		best = min(best, calc(ls, tree[rs].ele[i], dgt - 1));
		ans += best + (1 << dgt);
	}
}
int main(void)
{
	in(n);
	for (R int i = 1; i <= n; ++i)
	{
		in(val[i]);
		insert(root, val[i], 30);
	} 
	DFS(root, 30);
	printf("%lld\n", ans);
}