最长异或值路径(Trie)

给定一个树，树上的边都具有权值。

树中一条路径的异或长度被定义为路径上所有边的权值的异或和：

给定上述的具有 n 个节点的树，你能找到异或长度最大的路径吗？

输入格式
第一行包含整数 n，表示树的节点数目。

接下来 n−1 行，每行包括三个整数 u，v，w，表示节点 u 和节点 v 之间有一条边权重为 w。

输出格式
输出一个整数，表示异或长度最大的路径的最大异或和。

数据范围
1≤n≤100000,
0≤u,v<n,
0≤w<2^31
输入样例：
4
0 1 3
1 2 4
1 3 6
输出样例：
7
样例解释
样例中最长异或值路径应为 0->1->2，值为 7(=3⊕4)

• 题目是求任意两个节点路径的异或长度最大值，可以转化为先求每个点到根节点的异或长度，那么就转变为了求A1.A2…Ai中任意两个数的异或最大值问题
• 求每个点到根节点的距离深搜，由于是无向图，避免在搜索过程中往回搜，因此传入个父节点的参数

C++代码：

#include <iostream>
#include <cstring>

using namespace std;

const int N = 1e5 + 10;

int n;
int h[N],w[2*N],e[2*N],ne[2*N],cnt;
int son[31*N][2],idx;
int a[N];

void add(int a,int b,int c){  //邻接表存储树
    e[cnt] = b; w[cnt] = c; ne[cnt] = h[a]; h[a] = cnt++;
}

void insert(int x){  //字典树存储
    int p = 0;
    for(int i = 30; i >= 0; i--){
        int u = x >> i & 1;
        if(!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }
}

int query(int x){  //字典树查询
    int p = 0, res = 0;
    for(int i = 30; i >= 0; i--){
        int u = x >> i & 1;
        if(son[p][!u]){
            p = son[p][!u];
            res = res * 2 + !u;
        }else{
            p = son[p][u];
            res = res * 2 + u;
        }
    }
    return res;
}

void dfs(int u,int father,int sum){  //深搜，遍历每个点到根节点的异或值
    a[u] = sum;
    for(int i = h[u]; ~i; i = ne[i]){
        int j = e[i];
        if(j != father) dfs(j,u,sum ^ w[i]);
    }
}

int main(){
    scanf("%d",&n);
    memset(h,-1,sizeof h);
    for(int i = 0; i < n - 1; i++){
        int u,v,c;
        scanf("%d%d%d",&u,&v,&c);
        add(u,v,c);
        add(v,u,c);
    }
    
    dfs(0,-1,0);
    
    for(int i = 0; i < n; i++) insert(a[i]);
    int res = 0;
    for(int i = 0; i < n; i++) res = max(res,a[i]^query(a[i]));
    printf("%d\n",res);
    return 0;
}