《算法竞赛进阶指南》最长异或值路径

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

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

⊕⊕ 为异或符号。

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

输入格式

第一行包含整数 nn，表示树的节点数目。

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

输出格式

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

数据范围

1≤n≤1000001≤n≤100000,
0≤u,v<n0≤u,v<n,
0≤w<2310≤w<231

输入样例：

4
0 1 3
1 2 4
1 3 6

输出样例：

7

样例解释

样例中最长异或值路径应为 0->1->2，值为 7(=3⊕4)

 

题意：取树中任意两点使得该两点之间路径的异或值最大：

解题思路：

 

如图所示，要求a, b之间的异或值即求得a到root(根节点)的异或值，b到root的异或值分别即为f[a, root], f[b, root]。则a, b之间路径的值为sum = f[a, root] + f[b, root] - 2 * f[c][root]因为加减法可以转化为异或运算即：sum == f[a, root] ^ f[b, root] ^ f[c, root] ^ f[c, root] == f[a, root[ ^ f[b, root];因此题目转化为求两个数的异或值得最大值只需要将每个点到更节点的距离确定，再将其放入trie树中即可。具体课参考这篇博客：(2条消息) 《算法竞赛进阶指南》最大异或对_wsh1931的博客-CSDN博客

代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 100010, M = 2 * N;

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

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

void dfs(int u, int fa, int sum)//u为当前递归得点，fa为u是从哪个点过来的（防止往回搜）,
                                //sum为当前点到根节点得异或值
{
    a[u] = sum;
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (j != fa) dfs(j, u, sum ^ w[i]);//若没有往回搜，则继续递归到下一层
    }
}

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])
        {
            res += 1 << i;
            p = son[p][!u];
        }
        else
        {
            p = son[p][u];
        }
    }
    
    return res;
}

int main()
{
    cin >> n;
    memset(h, -1, sizeof h);
    for (int i = 0; i < n - 1; i ++ )
    {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        add(a, b, c), add(b, a, c);//无向图
    }
    
    dfs(0, -1, 0);//计算数中每个点到根节点得距离，这里得根节点可以是任一点，我这里选择0为根节点
    
    for (int i = 0; i < n; i ++ ) insert(a[i]);//将每个点到根节点得异或值插入trie树
    
    int res = 0;
    for (int i = 0; i < n; i ++ ) res = max(res, query(a[i]));//求出两个异或值最大得数
    
    cout << res << endl;
    
    return 0;
}