P4551 最长异或路径 trie树

  

一般遇到异或  很多题目需要构建01trie 

这题只要构建01树  将n个点每个点到根节点(1)的异或路径放入字典树    

然后再对这n条到根节点的异或路径求最大值即可

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i>=(b);--i)
#define ll long long
#define see(x) (cerr<<(#x)<<'='<<(x)<<endl)
#define inf 0x3f3f3f3f
#define CLR(A,v)  memset(A,v,sizeof A)
//////////////////////////////////
const int N=2e6+10;
struct Edge{int to,v,nex;}edge[N<<1];int head[N],pos,root;
void add(int a,int b,int c){edge[++pos]=(Edge){b,c,head[a]};head[a]=pos;}
int t[N][2],tot,bian[N];
void upnode(int x,int rt)
{
    for(int i=(1<<30);i;i>>=1)
    {
        bool c=i&x;
        if(!t[rt][c])t[rt][c]=++tot;
        rt=t[rt][c];
    }
}
int qmax(int x,int rt)
{
    int ans=0;
    for(int i=(1<<30);i;i>>=1)
    {
        bool c=i&x;
        if(t[rt][c^1])ans+=i,rt=t[rt][c^1];
        else rt=t[rt][c];
    }
    return ans;
}
int n,m,x,y,z;

void dfs(int x,int fa)
{
    for(int i=head[x];i;i=edge[i].nex)
    {    
        int v=edge[i].to;
        if(v==fa)continue;
        bian[v]=bian[x]^edge[i].v;
        dfs(v,x);
    }
}
int main()
{    
    cin>>n;
    rep(i,1,n-1)
    {    
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);
    }
    dfs(1,0);
    rep(i,1,n)upnode(bian[i],root);
    int ans=0;
    rep(i,1,n)ans=max(ans,qmax(bian[i],root));
    cout<<ans;
    return 0;
}
View Code

猜你喜欢

转载自www.cnblogs.com/bxd123/p/11511573.html