洛谷 P4551 最长异或路径（01字典树+dfs）（好题）

题目链接

思路：求最大异或我们考虑01字典树，但是在树上怎么建呢？我们考虑一下树上一点u，点u到其他点的最大异或和路径可以通过边dfs边建字典树的方式进行。

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=2e5+5;
typedef long long ll;
vector<pair<int,int> >g[maxn];
ll tree[maxn*32][2],tot,ans;
void insert(ll x)
{
	int u=0;
	for(int i=30;i>=0;--i)
	{
		int t=(((1LL<<i)&x)?1:0);
		if(!tree[u][t]) tree[u][t]=++tot;
		u=tree[u][t];
	}
}
ll query(ll x)
{
	int u=0;
	ll res=0;
	for(int i=30;i>=0;--i)
	{
		int t=(((1LL<<i)&x)?1:0);
		if(tree[u][t^1]) u=tree[u][t^1],res+=(1LL<<i);
		else u=tree[u][t];
	}
	return res;
}
void dfs(int x,int fa,ll sum)
{
	ans=max(ans,query(sum));
	for(int i=0;i<g[x].size();++i)
	{
		int to=g[x][i].first;
		if(to==fa) continue;
		ll t=sum^g[x][i].second;
		insert(t);
		dfs(to,x,t);
	}
}
int main()
{
	int n,u,v,w;
	scanf("%d",&n);
	for(int i=0;i<n-1;++i)
	{
		scanf("%d %d %d",&u,&v,&w);
		g[u].push_back({v,w});
		g[v].push_back({u,w});
	}
	insert(0);
	dfs(1,-1,0);
	printf("%lld\n",ans);
}