Luogu P4551 最长异或路径

题目链接 \(Click\) \(Here\)

\(01Trie\)好题裸题。

取节点\(1\)为根节点，向下扫每一个点从根节点到它路径上的异或和，我们可以得到一个\(sumx[u]\)。

现在路径异或和有两类：

• 跨过根节点，这种的异或路径长度等于两个子节点的\(sumx\)异或和异或起来的数值大小
• 在一棵子树中，这种的异或路径等于\(sumx[u]\)异或上\(sumx[v]\)再异或掉两次\(sumx[1->lca (u, v)]\)（因为被额外计算），依然等于两个子节点的\(sumx\)异或和异或起来的数值大小。

所以问题转为求在\(sumx\)中，对每个\(sumx[u]\)，和它产生最大异或和的\(sumx[v]\)最大可以异或出来多少。我们把数列每个值插入\(01Trie\)中，求解时尽可能选择对应位不同的数。复杂度\(O(N*31)\)

#include <bits/stdc++.h>
using namespace std;

const int N = 100010;

int cnt, head[N];

struct edge {
	int nxt, to, w;
}e[N << 1];

void add_edge (int from, int to, int val) {
	e[++cnt].nxt = head[from];
	e[cnt].to = to;
	e[cnt].w = val;
	head[from] = cnt;
}

void add_len (int u, int v, int w) {
	add_edge (u, v, w);
	add_edge (v, u, w);
}

int n, u, v, w, sumx[N];

void get_sumx (int u, int fa) {
	for (int i = head[u]; i; i = e[i].nxt) {
		int v = e[i].to;
		if (v != fa) {
			sumx[v] = sumx[u] ^ e[i].w;
			get_sumx (v, u);
		}
	}
}


int ch[N * 31][2], max_size;

void insert (int val) {
	int now = 0;
	for (int i = 30, to = 0; i >= 0; --i) {
		to = ((val & (1 << i)) != 0); //如果 val 第 i 位上为 1
		if (!ch[now][to]) {
			ch[now][to] = ++max_size;
		}
		now = ch[now][to];
	}
}

int get_ans (int val) {
	int now = 0, ans = val;
	for (int i = 30; i >= 0; --i) {
		if (ans & (1 << i)) {
			//这一位为1 -> 向0走
			if (ch[now][0] != 0) {
				now = ch[now][0];
			} else {
				val ^= (1 << i);
				now = ch[now][1];
			}
		} else {
			//为0 -> 向1走
			if (ch[now][1] != 0) {
				val ^= (1 << i);
				now = ch[now][1];
			} else {
				now = ch[now][0];
			}
		}
	}
	return max (val, ans);
}

int main () {
	cin >> n;
	for (int i = 1; i <= n - 1; ++i) {
		cin >> u >> v >> w;
		add_len (u, v, w);
	}
	get_sumx (1, 0);
	for (int i = 1; i <= n; ++i) {
		insert (sumx[i]);
	}
	int ans = 0;
	for (int i = 1; i <= n; ++i) {
		ans = max (ans, get_ans (sumx[i])); //求sumx与其他数的最大异或
	}
	cout << ans << endl;
}