[LGOJ4551]最长异或路径——[01trie]

【题目描述】

给定一棵n个点的带权树，结点下标从1开始到N。寻找树中找两个结点，求最长的异或路径。

异或路径指的是指两个结点之间唯一路径上的所有边权的异或。

【输入格式】

第一行一个整数N，表示点数。

接下来 n-1行，给出 u,v,w ，分别表示树上的 u 点和 v 点有连边，边的权值是 w。

【输出格式】

一行，一个整数表示答案。

$SampleInput$

4
1 2 3
2 3 4
2 4 6

$SampleOutput$

7

【题意分析】

今天%你赛考到这道题，太菜了没想到01trie，然后爆零了

我们记prefix[x]表示从根节点到x路径的异或和

那么很显然路径(u,v)的异或和就是$prefix[u]$^$prefix[v]$（自己动手画画就知道了）

但是我们这样做还要$O(n^2)$处理答案，tle

怎么优化呢？考虑对每次dfs时从根到当前节点的异或和全都扔进一棵01trie里面

这样操作复杂度是$O(可行)$

我们可以统计总共插入了多少单词（异或和），对于每一种异或和（提前用数组存起来），在01trie上query

实现方法：从根节点（0）开始一步一步往下找，如果存在不一样的儿子，走这个儿子（如果有），没有的话，就走相同儿子递归下去。

为什么存在可以不一样就要走那个儿子？因为从上往下找，走不一样的儿子就意味着这一位是1而不是0，那么结果也就更大。

code：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
#define MAXN 1000010
using namespace std;

struct fls {
	int to, next, val;
}edge[MAXN << 1];

int son[MAXN * 30][2], a[MAXN], head[MAXN << 1], n, ans, cnt, tot, sz;

inline int read () {
	register int s = 0, w = 1;
	register char ch = getchar ();
	while (! isdigit (ch)) {if (ch == '-') w = -1; ch = getchar ();}
	while (isdigit (ch)) {s = (s << 3) + (s << 1) + (ch ^ 48); ch = getchar ();}
	return s * w;
}

inline void connect (int u, int v, int w) {
	edge[++cnt].to = v, edge[cnt].val = w, edge[cnt].next = head[u], head[u] = cnt;
}

struct trie {
	
	inline void insert (int x) {
		int u = 0;
		for (register int i = 30; i >= 0; i--) {
			int c = (x >> i) & (long long) 1;
			if (! son[u][c]) son[u][c] = ++sz;
			u = son[u][c];
		}
	}
	
	inline int query (int x) {
		int u = 0, res = 0;
		for (register int i = 30; i >= 0; i--) {
			int c = (x >> i) & (long long) 1, flag = 0;
			if (son[u][c ^ 1]) res |= (1 << i), flag = 1;
			u = flag ? son[u][c ^ 1] : son[u][c];
		}
		return res;
	}
	
}tree;

void DFS (int now, int fa, int sum) {
	a[++tot] = sum, tree.insert (sum);
	for (register int i = head[now]; i; i = edge[i].next) {
		int v = edge[i].to; if (v == fa) continue;
		DFS (v, now, sum ^ edge[i].val);
	}
}

int main () {
	n = read ();
	for (register int i = 1; i < n; i++) {
		int u = read (), v = read (), w = read ();
		connect (u, v, w), connect (v, u, w);
	}
	DFS (1, 0, 0);
	for (register int i = 1; i <= tot; i++) ans = max (ans, tree.query (a[i]));
	printf ("%lld\n", ans);
	return 0;
}