POJ 3764 The xor-longest Path trie树解决位运算贪心

http://poj.org/problem?id=3764

题意 :  一颗树，每个边有个值，在树上找一条简单路径，使得这条路径上的边权异或值最大

先找到所有节点到一点的距离 , 显然dis( x , y ) = dis( z , x )^dis( z , y )

那么把所有的距离都以二进制由高到低存到trie中 , 扫一遍每个点 , 在trie树上由高到低找某位与该点某位相反的数 , 找不到则妥协找下一位 , 如此贪心即可 

写的时候超空间一次re两次,trie树的大小设置为30*n刚好,不要想太多设置太大或者因为走极端设置得过小...就按数据量设置就可以...

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=100010;
const double eps=1e-8;
const long long modn=1000;
int n;
int ans=0;
struct nod{
    int y;
    int next;
    int v;
}e[maxn*2];
int head[maxn]={},tot=0;
int dis[maxn]={};
struct tri{
    bool exist;
    int v;
    int next[2];
}a[maxn*30];
int tot1=0;
void init(int x,int y,int z){
    e[++tot].y=y;
    e[tot].v=z;
    e[tot].next=head[x];
    head[x]=tot;
}
void doit(int x,int v,int k){
    if(k<0){
        a[x].exist=1;
        a[x].v=v;
        return;
    }int z;
    if(v&(1<<k)){
        z=1;
    }else{
        z=0;
    }
    if(!a[x].next[z]){
        a[x].next[z]=++tot1;
    }doit(a[x].next[z],v,k-1);
}
void getit(int x,int v,int k){
    if(k<0){
        int z=v^a[x].v;
        ans=max(ans,z);
        return;
    }int z;
    if(v&(1<<k)){
        z=0;
    }else{
        z=1;
    }
    if(!a[x].next[z]){
        z=1^z;
    }getit(a[x].next[z],v,k-1);
}
void dfs(int x,int fa,int val){
    int y,v,z;
    for(int i=head[x];i;i=e[i].next){
        y=e[i].y;
        v=e[i].v;
        if(y!=fa){
            z=val^v;
            dis[y]=z;
            dfs(y,x,z);
        }
    }
}
void yu(){
    memset(a,0,sizeof(a));
    memset(e,0,sizeof(e));
    memset(head,0,sizeof(head));
    tot1=tot=0;
    memset(dis,0,sizeof(dis));
    ans=0;
}
int main(){
    while(~scanf("%d",&n)){
        yu();
        int x,y,z;
        for(int i=1;i<n;i++){
            scanf("%d%d%d",&x,&y,&z);
            init(x+1,y+1,z);
            init(y+1,x+1,z);
        }
        dfs(1,0,0);
        for(int i=1;i<=n;i++){
            getit(0,dis[i],30);
            doit(0,dis[i],30);
        }
        printf("%d
",ans);
    }
    return 0;
}