AcWing 144. 最长异或值路径（Trie字典树）

AcWing 144. 最长异或值路径

https://www.acwing.com/problem/content/146/

题目大意

给定一棵 N 个节点的树，树上的每条边都有一个权值。从树中选择两个点 x 和y，把从 x 到 y 的路径上的所有边权 xor(异或)起来，得到的结果最大是多少?（N <= 10^5）

输入输出

输入一个数字N，然后输入N - 1行数字，每行包括三个整数 u，v，w，表示节点 u 和节点 v 之间有一条边权重为 w。

样例输入：

4
0 1 3
1 2 4
1 3 6

样例输出：

7

思路

题目中将树中一条路径的异或长度定义为路径上所有边的权值的异或和，题目要求求出树上x到y的路径上的所有边权的xor（异或）结果。x到y的路径上的所有边权的xor结果相当于**x到根节点的所有边权的xor值再 xor y到根节点的所有边权的xor值。**也就相当与一个差分的思想，只不过是在树上进行操作。

设D[x]表示根节点到x的路径上的所有边权的xor值， 则我们可以得出：
D[x] = D[father(x)] xor weight(x, father(x))

根据上式，我们可以对树进行一次深度优先遍历，求出所有的D[x]。这一步需要我们先将输入的所有节点和权值用链式向前星（数组实现的邻接表），然后在进行dfs遍历。

而树上x到y的路径上的所有边权的xor结果就等于D[x] xor D[y]。这是因为根据xor运算的性质（a xor a = 0），“x 到根”和“y 到根”这两条路径重叠的部分恰好抵消掉。

所以，问题就变成了从 D[1]~D[N] 这N个数中选出两个，xor 的结果最大，即上一道例题。可以用 Trie 树来快速求解。

C++实现

//AcWing 144. 最长异或值路径
/**/
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>

using namespace std;

#define IOS ios::sync_with_stdio(false); cin.tie(0), cout.tie(0);
#define ll long long
#define endl '\n'

typedef pair<char, int> pir;
const int mod = 0x7f7f7f7f;
const int N = 100010;

int trie[N * 31][2], d[N], c;
int ver[N * 2], e[N * 2], ne[N * 2], h[N], tot;

void add(int x, int y, int z){
	ver[tot] = y;
	e[tot] = z;
	ne[tot] = h[x];
	h[x] = tot++;
}

void dfs(int x, int f, int sum){
	d[x] = sum;
	for(int i = h[x]; ~i; i = ne[i]){
		int j = ver[i];
		if(j != f) dfs(j, x, sum ^ e[i]);
	}
}

void insert(int x){
	int q = 0;
	for(int i = 30; i >= 0; i -- ){
		int u = x >> i & 1;
		if(!trie[q][u]) trie[q][u] = ++ c;
		q = trie[q][u];
	}
}

int query(int x){
	int q = 0, ans = 0;
	for(int i = 30; i >= 0; i -- ){
		int u = x >> i & 1;
		if(trie[q][!u]){
			ans += 1 << i;
			q = trie[q][!u];
		} else {
			q = trie[q][u];
		}
	}
	return ans;
}

int main(void){
    IOS
	memset(h, -1, sizeof h);
    int n; cin >> n;
	for(int i = 1; i < n; i ++ ){
		int x, y, z; cin >> x >> y >> z;
		add(x, y, z); add(y, x, z);
	}
	
	dfs(0, -1, 0);
	
	for(int i = 1; i <= n; i ++ ){
		insert(d[i]);
	}
	int res = 0;
	for(int i = 1; i <= n; i ++ ){
		res = max(res, query(d[i]));
	}
	cout << res << endl;
	
    return 0;
}