洛谷P4551 最长异或路径 01trie

题目传送门

Description

给你一棵有边权的树，求最长异或路径。

Solution

和这题基本没区别。

不妨令 $p_i$ 为 $i$ 号节点到一号的路径异或值。

则点 $u$ 和点 $v$ 的路径异或值为 $p_u\oplus p_v$。

Code

#include<bits/stdc++.h>
using namespace std;
int n,a[105000],vis[105000];
int head[105000],cnt;
struct edge{
	int v,w,next;
}e[205000];
void add(int u,int v,int w){
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}
void dfs(int u,int sum){
	vis[u]=1;a[u]=sum;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].v;
		if(vis[v]==0)dfs(v,a[u]^e[i].w);
	}
}
int trie[10500000][2],tot=0,ans=-1;
void insert(int x){
	int p=0;
	for(int i=31;i>=0;--i){
		int c=(x>>i)&1;
		if(trie[p][c]==0)trie[p][c]=++tot;
		p=trie[p][c];
	}
}
int query(int x){
	int p=0,sum=0;
	for(int i=31;i>=0;--i){
		int c=(x>>i)&1;
		if(trie[p][!c]){
			p=trie[p][!c];
			sum=(sum<<1)|1;
		}
		else{
			p=trie[p][c];
			sum=sum<<1;
		}
	}
	return sum;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n-1;++i){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);add(v,u,w);
	}
	dfs(1,0);
	for(int i=1;i<=n;++i)insert(a[i]);
	for(int i=1;i<=n;++i)ans=max(ans,query(a[i]));
	printf("%d\n",ans);
}