LOJ The XOR-longest Path

The XOR-longest Path

题目描述见上面链接

题解：
trie树专题，那么就往这上面想想。

每一条路的路径长度都可以转为二进制用 trie 树存下。

这样的话，我们从头开始，只需比较每次左右 0 和 1 两个儿子都存在，则对答案的贡献则为 1<<w （w 为位置），否则对答案的贡献就为 0 。不得不说真是巧妙啊！

贴代码：

#include<bits/stdc++.h>
#define N 100005
#define M 3100005
using namespace std;
int ans,n,tot,num,nxt[N*2],w[N*2],to[N*2],hea[N],f[N],trie[M][2];
void insert(int x)
{
    int rt=0,w=30;
    while (w>=0)
    {
        int y=(x>>w) & 1;
        if (!trie[rt][y]) trie[rt][y]=++num;
        rt=trie[rt][y]; 
        w--;
    }
}
void find(int x)
{
    int sum=0,w=30,rt=0;
    while (w>=0)
    {
        int y=(x>>w) & 1;
        if (trie[rt][!y]) sum+=1<<w,rt=trie[rt][!y];
        else rt=trie[rt][y];
        w--;
    }
    ans=max(ans,sum);
}
void add(int x,int y,int z)
{
    nxt[++tot]=hea[x]; hea[x]=tot; w[tot]=z; to[tot]=y;
}
void dfs(int x,int fa)
{
    find(f[x]); insert(f[x]);
    for (int j=hea[x]; j; j=nxt[j])
      if (to[j]!=fa) f[to[j]]=f[x]^w[j],dfs(to[j],x);
}
int main()
{
    scanf("%d",&n);
    for (int i=1,x,y,z; i<n; i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z); add(y,x,z);
    }
    dfs(1,0);
    printf("%d",ans);
    return 0;
}

加油加油加油！！！fighting fighting fighting!!!