CF888G Xor-MST DFS 最小生成树 01Trie

BestMonkey

已于 2024-01-19 15:42:13 修改

阅读量914

点赞数 27

分类专栏：题解文章标签： c++ c语言算法

于 2024-01-19 11:18:34 首次发布

本文链接：https://blog.csdn.net/bestmonkey/article/details/135692143

版权

题解专栏收录该内容

63 篇文章 0 订阅

订阅专栏

Xor-MST

传送门

题面翻译

给定 $n$ 个结点的无向完全图。每个点有一个点权为 $a_i$ 。连接 $i$ 号结点和 $j$ 号结点的边的边权为 $a_i\oplus a_j$ 。
求这个图的 MST 的权值。
$1\le n\le 2\times 10^5$ ， $0\le a_i< 2^{30}$ 。

题目描述

You are given a complete undirected graph with $n$ vertices. A number $a_{i}$ is assigned to each vertex, and the weight of an edge between vertices $i$ and $j$ is equal to $a_{i}xora_{j}$ .

Calculate the weight of the minimum spanning tree in this graph.

输入格式

The first line contains $n$ ( $1 <= n <= 200000$ ) — the number of vertices in the graph.

The second line contains $n$ integers $a_{1}$ , $a_{2}$ , …, $a_{n}$ ( $0<=a_{i}<2^{30}$ ) — the numbers assigned to the vertices.

输出格式

Print one number — the weight of the minimum spanning tree in the graph.

样例 #1

样例输入 #1

5
1 2 3 4 5

样例输出 #1

样例 #2

样例输入 #2

4
1 2 3 4

样例输出 #2

$以上来自洛谷$

解题思路

先讲 $B or uv ka$ 算法：当前每个集合伸出去一条最短的边，然后把联通块缩成一个新的集合，用这种思想处理 $MST$ 。（学习 $MST$ 看这里）
但是如果我们要使得异或和最小，这个很容易想到 $01 T r i e$ （学习 $T r i e$ 树看这里），考虑将每一个点按照权值插到 $T r i e$ 中，这样我们考虑叶子节点就是实际存在的点，而非叶子节点一定时若干对叶子节点的 $L C A$ 。（不知道 $L C A$ ？看这里）
根据二进制的性质，不难发现我们先合并 $L C A$ 深度大的点对更优，这样我们考虑 $\operatorname{dfs}$ 整个 $T r i e$ ，对于一个非叶子节点，我们枚举他的 $0/1$ 子树的所有儿子，然后在 $1/0$ 子树中找到边权最小的来合并，如果我们从小到大插入，那么编号就是连续的，然后启发式合并即可。

AC Code

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int Maxn = 6e6 + 5;
int n, a[Maxn];
int rt, L[Maxn], R[Maxn], ch[Maxn][2], tot;
inline void Insert(int &now, int x, int dep) {
	if (!now) {
		now = ++tot;
	}
	L[now] = min(L[now], x), R[now] = max(R[now], x);
	if (dep < 0) {
		return;
	}
	int bit = a[x] >> dep & 1;
	Insert(ch[now][bit], x, dep - 1);
}
inline int query(int now, int val, int dep) {
	if (dep < 0) {
		return 0;
	}
	int bit = val >> dep & 1;
	if (ch[now][bit]) {
		return query(ch[now][bit], val, dep - 1);
	} else {
		return query(ch[now][bit ^ 1], val, dep - 1) + (1 << dep);
	}
}
inline int dfs(int now, int dep) {
	if (dep < 0) {
		return 0;
	}
	if (R[ch[now][0]] && R[ch[now][1]]) {
		int minn = INT_MAX;
		for (int i = L[ch[now][0]]; i <= R[ch[now][0]]; i++) {
			minn = min(minn, query(ch[now][1], a[i], dep - 1));
		}
		return dfs(ch[now][0], dep - 1) + dfs(ch[now][1], dep - 1) + minn + (1 << dep);
	}
	if (R[ch[now][0]]) {
		return dfs(ch[now][0], dep - 1);
	}
	if (R[ch[now][1]]) {
		return dfs(ch[now][1], dep - 1);
	}
	return 0;
}
inline void work() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	sort(a + 1, a + n + 1);
	memset(L, 0x3f, sizeof(L));
	for (int i = 1; i <= n; i++) {
		Insert(rt, i, 30);
	}
	cout << dfs(rt, 30) << endl;
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	work();
	return 0;
}