一道trie的经典例题
先dfs一遍某一个点到根节点的亦或和
然后通过这些亦或和建立一颗trie数组
然后贪心走一条亦或和最大的路径
#include<iostream>
#include<fstream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=1e5+3;
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&i;
}
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&i;
}
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(){
cin>>n;
for(int i=1;i<n;i++){
int u,v,w;
cin>>u>>v>>w;
g[u].push_back((edge){u,v,w});
g[v].push_back((edge){v,u,w});
}//加边
dfs(1,1);//dfs一遍某一个点到根的亦或和
int ans=0;
for(int i=1;i<=n;i++){
insert(a[i]);
}//建立一颗二进制的trie
for(int i=1;i<=n;i++){
ans=max(ans,a[i]^qmax(a[i]));
}
cout<<ans<<endl;
return 0;
}
/*
in:
4
1 2 3
2 3 4
2 4 6
out:
7
*/