BZOJ P1954: Pku3764 The xor-longest Path

一道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
*/