【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;
}