【洛谷P4551】【ybtoj 2.4】C.最长异或路径【trie】

ssllyr

已于 2022-08-16 14:26:58 修改

阅读量162

点赞数

分类专栏： trie YbtOJ 文章标签： trie

于 2021-07-09 14:41:55 首次发布

本文链接：https://blog.csdn.net/dglyr/article/details/118606493

版权

YbtOJ 同时被 2 个专栏收录

69 篇文章 0 订阅

订阅专栏

trie

5 篇文章 0 订阅

订阅专栏

本文介绍了一种算法，通过预处理从根节点到每个节点的异或路径权重，并利用深度优先搜索（DFS）收集这些值。核心在于理解异或性质在解决最大异或问题中的应用。代码展示了如何构造 Trie 数据结构进行辅助计算。

摘要由CSDN通过智能技术生成

在这里插入图片描述

分析

预处理出从根节点到每一个节点路径上所有边权的异或值。因为同一个数异或起来就是0。其实不同的位越靠前就越大。
然后就是最大异或对那道题了

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int n,a[200020];
int hd[200020],tot;
int trie[3200020][10];
long long ans;

struct lwx
{
    int to,next,w;
}e[200010];

void add(int x,int y,int w)
{
    e[++tot]=(lwx){y,hd[x],w};
    hd[x]=tot;
}

void dfs(int x,int fa)
{
    for(int i=hd[x];i>0;i=e[i].next)
    {
        if(e[i].to==fa) continue;
        a[e[i].to]=a[x]^e[i].w;
        dfs(e[i].to,x);
    }
}

void in(int t)
{
    int now=1;
    for(int i=31;i>=0;i--)
    {
        int ch=(t>>i)&1;
        if(!trie[now][ch])
        {
            trie[now][ch]=++tot;
        }
        now=trie[now][ch];
    }
}

void fd(int t)
{
    long long now=1,s=0;
    for(int i=31;i>=0;i--)
    {
        int ch=!((t>>i)&1);
        if(trie[now][ch]) s+=(1<<i);
        else ch=!ch;
        now=trie[now][ch];
    }
    ans=max(ans,s);
}

int main()
{
    cin>>n;
    for(int i=1;i<=n-1;i++)
    {
        int x,y,w;
        cin>>x>>y>>w;
        add(x,y,w);
        add(y,x,w);
    }   
    dfs(1,0);
    for(int i=1;i<=n;i++)
    {
        fd(a[i]);
        in(a[i]);
    }
    cout<<ans;
    return 0;
}