传送门:POJ 3764
题目大意:
在树上找一段路径(连续)使得边权相异或的结果最大。
前置技能:
-
用链式前向星建图。
-
01字典树的应用。
思路:
本题用 vector数组建图是会超时的,所以只能用链式前向星建图了。
由于是求异或的最值问题,可以考虑用 01字典树解决,又因为是在树上进行的,所以可以在 DFS遍历树的节点的同时把 当前节点到根节点相异或的结果 在01字典树中求异或的最大值并加到01字典树中去。
你可能好奇,DFS过程中并没有取消点的访问,这样为什么就能保证你所求的结果对应的路径是连续的呢?考虑三种情况:
-
深搜的两条路径一个包含在另一个中,这样一异或,公共的部分的值变成 0 了,剩下的部分是连续的。
-
深搜的两条路径部分重叠,同理,异或后公共的部分变成 0 了,由于是对边权的异或,剩下的部分仍然连续。
-
深搜的两条路径不重叠,但是两条路径肯定有公共的点——根节点,也是连续的。
代码
#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;
}