POJ 3764 The xor-longest Path 01字典树 树上最大异或和

题目链接

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

题意

给定一个N个结点、N-1条边的树。每个边上有一个权值。求树上路径权值的最大异或和。

解题

以零号结点为根进行一遍dfs处理出结点u到根节点的路径异或和。然后将这些异或和插入到01字典树中。最后遍历这些异或和查询与其异或最大的路径异或和。

昨天写这题，怎么写怎么WA。今天起来重写一遍，怎么写怎么AC。真的迷啊。突然发现debug的神招，第二天再写或者重写。orz。
我想如果张爱玲是个程序员的话，她一定会说，debug只需要两样东西：时间和新欢。（逃

AC代码

//688ms
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;

const int maxn=1e5+7;

int ch[32*maxn][2],sz;
int val[32*maxn];
struct edge
{
    int to,next,w;
}e[maxn<<1];
int head[maxn],cnt;
void init()
{
    memset(ch[0],0,sizeof(ch[0]));
    memset(head,-1,sizeof(head));
    cnt=-1;
    sz=0;
}
void insert(int x)
{
    int p=0;
    for(int i=31;i>=0;i--)
    {
        int c=(x>>i)&1;
        if(!ch[p][c])
        {
            ch[p][c]=++sz;
            val[sz]=0;//?
            memset(ch[sz],0,sizeof(ch[sz]));
        }
        p=ch[p][c];
    }
    val[p]=x;
}
int query(int x)
{
    int p=0;
    for(int i=31;i>=0;i--)
    {
        int c=(x>>i)&1;
        if(ch[p][c^1]) p=ch[p][c^1];
        else p=ch[p][c];
    }
    return val[p]^x;
}


void add_edge(int u,int v,int w)
{
    e[++cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}
int ans,pre[maxn],vis[maxn];
void dfs(int u)
{
    vis[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int v=e[i].to;
        if(vis[v]) continue;
        pre[v]=pre[u]^e[i].w;
        dfs(v);
    }
}

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        ans=0;
        init();
        for(int i=1;i<n;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add_edge(u,v,w);
            add_edge(v,u,w);
        }
        memset(vis,0,sizeof(vis));
        pre[0]=0;
        dfs(0);

        for(int i=0;i<n;i++)
           // printf("%d\n",pre[i]);
            insert(pre[i]);
        for(int i=1;i<n;i++)
            ans=max(ans,query(pre[i]));
        printf("%d\n",ans);
    }
    return 0;
}