欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

The xor-longest Path POJ - 3764(树上异或最长路径+trie)

程序员文章站 2022-04-17 21:37:30
...

题意:传送门
题解:给出一棵树,求解树上一条路径异或和最大,首先考虑跑一个有根树,d[i]d[i]表示从ii到根的异或和,之后比如iijj的路径异或和就是d[i]d[j]d[i]\wedge d[j],这点可以推算得知,从一堆数中选出两个数异或和最大,暴力O(n2)O(n^2),但是如果优化下,从第3131位开始想,如果为11,那么从剩下数中看第3131位有是00的么,那么再从这些依次选取,否则就只好选11的那部分,之后方法类似,可以把这些数都建成trietrie树,然后从3131位到11位跑,复杂度O(n)31O(n)*31,这样就可以得到正解了。
ccode:ccode:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1e5+5;
const int M=3e6;
int son[M][2],idx,n,a[N],u,v,w,res;
struct edge{
    int u,v,w,next;
}edges[N<<2];
int head[N],tot;
void add_edge(int u,int v,int w)
{
    edges[tot].u=u;edges[tot].v=v;edges[tot].w=w;edges[tot].next=head[u];head[u]=tot++;
    edges[tot].u=v;edges[tot].v=u;edges[tot].w=w;edges[tot].next=head[v];head[v]=tot++;
}
void dfs(int u,int f,int sum)
{
    a[u]=sum;
    for(int i=head[u];~i;i=edges[i].next){
        if(edges[i].v!=f)dfs(edges[i].v,u,sum^edges[i].w);
    }
}
void insert(int x)
{
    int p=0;
    for(int i=30;~i;i--){
        int &s=son[p][x>>i&1];
        if(!s)s=++idx;
        p=s;
    }
}
int query(int x)
{
    int p=0,res=0;
    for(int i=30;~i;i--){
        int s=x>>i&1;
        if(son[p][!s]){
            res+=1<<i;
            p=son[p][!s];
        }else p=son[p][s];
    }
    return res;
}
int main()
{
    while(scanf("%d",&n)==1){
        memset(head,-1,sizeof(head));
        for(int i=0;i<n-1;i++){
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);
        }
        dfs(0,-1,0);
        idx=0;
        memset(son,0,sizeof(son));
        for(int i=0;i<n;i++)insert(a[i]);
        res=0;
        for(int i=0;i<n;i++){
            res=max(res,query(a[i]));
        }
        printf("%d\n",res);
    }
    return 0;
}

相关标签: trie