【BZOJ】【P1954】【Pku3764 The xor-longest Path】【题解】【Trie】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=1954

水题……3min写完1A……

Code:

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
struct edge{int u,v,w;};
vector<edge>G[maxn];
int n,m,a[maxn];
void dfs(int u,int fa){
	for(int i=0;i<G[u].size();i++){
		edge e=G[u][i];
		if(e.v==fa)continue;
		a[e.v]=a[u]^e.w;
		dfs(e.v,u);
	}
}
int ch[maxn<<4][2],tot;
short val[maxn<<4];
void insert(int x){
	int bit[32];
	for(int i=0;i<32;i++)bit[i]=x>>i&1;
	int u=0;
	for(int i=31;i>=0;i--){
		if(!ch[u][bit[i]])
		ch[u][bit[i]]=++tot;
		u=ch[u][bit[i]];	
	}val[u]=1;
}
int Qmax(int x){
	int bit[32];
	for(int i=0;i<32;i++)bit[i]=x>>i&1;
	int u=0,ans=0;
	for(int i=31;i>=0;i--){
		if(ch[u][!bit[i]]){
			ans|=(!bit[i])<<i;
			u=ch[u][!bit[i]];
		}else{
			ans|=bit[i]<<i;
			u=ch[u][bit[i]];
		}
	}return ans;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int u,v,w;scanf("%d%d%d",&u,&v,&w);
		G[u].push_back((edge){u,v,w});
		G[v].push_back((edge){v,u,w});
	}dfs(1,1);int ans=0;
	for(int i=1;i<=n;i++)insert(a[i]);
	for(int i=1;i<=n;i++)ans=max(ans,a[i]^Qmax(a[i]));
	cout<<ans<<endl;
	return 0;
}