bzoj1954: Pku3764 The xor-longest Path

传送门
首先假设1为根，求出根到当前节点的xor距离。
然后将xor距离转化为二进制串，建立Trie树。
强制一些点在答案上，然后在Trie树上跑贪心。
显然答案是正确的。
时间复杂度O（n*32）

#include<iostream>#include<cstdio>#include<algorithm>#include<cstring>#include<cmath>#include<cstdlib>#define N 100005using namespace std;struct edge{int to,next,v;}e[N*2];int head[N],d[N],a[N][32],c[N*32][2],bin[32];int n,x,y,z,tot,cnt,ans,top,sum;void add(int x,int y,int v){    e[++tot]=(edge){y,head[x],v};    head[x]=tot;}void dfs(int x,int fa,int v){    d[x]=v;    for (int i=head[x];i;i=e[i].next)        if (e[i].to!=fa)            dfs(e[i].to,x,v^e[i].v); }int main(){    scanf("%d",&n);    for (int i=1;i<n;i++){        scanf("%d%d%d",&x,&y,&z);        add(x,y,z);        add(y,x,z);    }    dfs(1,-1,0);    cnt=1; ans=0;    for (int i=bin[0]=1;i<=30;i++) bin[i]=bin[i-1]*2;    for (int i=1;i<=n;i++){        x=1; top=0;        for (;d[i];d[i]/=2,top++) a[i][top]=d[i]%2;        for (int j=30;j>=0;j--){            if (!c[x][a[i][j]]) c[x][a[i][j]]=++cnt;            x=c[x][a[i][j]];        }    }    for (int i=1;i<=n;i++){        sum=0,x=1;        for (int j=30;j>=0;j--){            if (c[x][a[i][j]^1]!=0) sum+=bin[j],x=c[x][a[i][j]^1];            else x=c[x][a[i][j]];        }        ans=max(ans,sum);    }    printf("%d",ans);}