POJ 3764 Language: The xor-longest Path （01字典树+DFS）

传送门：POJ 3764

题目大意：

在树上找一段路径（连续）使得边权相异或的结果最大。

前置技能：

1. 用链式前向星建图。

2. 01字典树的应用。

思路：

本题用 vector数组建图是会超时的，所以只能用链式前向星建图了。

由于是求异或的最值问题，可以考虑用 01字典树解决，又因为是在树上进行的，所以可以在 DFS遍历树的节点的同时把 当前节点到根节点相异或的结果 在01字典树中求异或的最大值并加到01字典树中去。

你可能好奇，DFS过程中并没有取消点的访问，这样为什么就能保证你所求的结果对应的路径是连续的呢？考虑三种情况：

1. 深搜的两条路径一个包含在另一个中，这样一异或，公共的部分的值变成 0 了，剩下的部分是连续的。

2. 深搜的两条路径部分重叠，同理，异或后公共的部分变成 0 了，由于是对边权的异或，剩下的部分仍然连续。

3. 深搜的两条路径不重叠，但是两条路径肯定有公共的点——根节点，也是连续的。

代码

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<vector>
#define MAXN 100010
using namespace std;
typedef long long LL;
 
struct node
{ //链式前向星节点 
	int to,next,val; //相邻节点，下一节点，边权 
} mp[MAXN*2];
 
int n,tol,ans;
int val[MAXN*32];   //01字典树点值 
int ch[MAXN*32][2]; //01字典树边的指向 
int cnt,head[MAXN]; //链式前向星 
 
void init()
{ //初始化 
	ans=0;
	cnt=0; 	
	tol=1;
	ch[0][0]=ch[0][1]=0;
	memset(head,-1,sizeof(head)); 
}
 
void add(int u,int v,int w)
{ //加边 
	mp[cnt].to=v;
	mp[cnt].val=w;
	mp[cnt].next=head[u];
	head[u]=cnt++;
}
 
void insert(int x)
{ //01字典树插入数字 
	int u=0;
	for(int i=31;i>=0;i--)
	{
		int v=(x>>i)&1;
		if(!ch[u][v])
		{ //如果节点未访问过，则创建 
			ch[tol][0]=ch[tol][1]=0; //边值为0表示不指向任何节点 
			val[tol]=0; //点值为0表示到此不表示一个数 
			ch[u][v]=tol++; //节点编号 
		}
		u=ch[u][v]; //下一节点 
	}
	val[u]=x; //点值 
}
 
int query(int x)
{ //01字典树查询与 x异或的最大值 
	int u=0;
	for(int i=31;i>=0;i--)
	{
		int v=(x>>i)&1;
		//按照贪心策略，优先选择和当前位不同的点 
		if(ch[u][v^1]) u=ch[u][v^1];
		else u=ch[u][v];
	}
	return x^val[u];
}
 
void dfs(int u,int pre,int val)
{ //参数：当前点，父节点，当前点到根节点的异或值 
	int i,v;
	insert(val);
	for(i=head[u];i!=-1;i=mp[i].next)
	{
		v=mp[i].to;
		if(v==pre) continue;
		//val=val^mp[i].val; //不可以这样做，因为会改变 val的值
		ans=max(ans,query(val^mp[i].val)); //查询结果的最大值 
		dfs(v,u,val^mp[i].val); //深搜 
	}
}
 
int main()
{
	int i,u,v,w;
	while(~scanf("%d",&n))
	{		
		init();
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			add(u,v,w);
			add(v,u,w);
		}				
		dfs(0,-1,0);
		printf("%d\n",ans);
	}
	return 0;
}