poj 3764 The xor-longest Path 字典树 + 前缀和

题目链接

题意：在一棵树上选择一个路径，使路径权值异或之和最大。
既然是在字典树专题里的，自然也就想到了字典树，不过只能求两个数的最大异或，而这个题不确定是不是两个数，而且路径也很难找出来，这个时候想起来之前学主席树做 “最大异或和” 这道模板题的时候，将区间异或转换成了前缀异或，茅塞顿开，发现可以用dfs处理前缀和，让后每一条路径不就是两个前缀异或起来嘛。这样就转换成了两个数的异或，可以用字典树轻松解决。
我不会算字典树的空间，每个题基本都得re一次，开的可能有点大。

#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define pb push_back
#define mk make_pair
using namespace std;

typedef long long LL;
typedef pair<int,int> PII;

const int N=100010,M=200020,S=50,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n,m;
int e[M],ne[M],w[M],h[N],idx;
int tr[N*S][2],tot;
vector<int>v;

void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void insert(int x)
{
    int p=0;
    for(int i=30;i>=0;i--)
    {
        int u=x>>i&1;
        if(!tr[p][u]) tr[p][u]=++tot;
        p=tr[p][u];
    }
}

void dfs(int u,int f,int sum)
{
    insert(sum),v.pb(sum);
    for(int i=h[u];~i;i=ne[i])
    {
        int j=e[i];
        if(j==f) continue;
        dfs(j,u,sum^w[i]);
    }
}

int query(int x)
{
    int p=0,ans=0;
    for(int i=30;i>=0;i--)
    {
        int u=x>>i&1;
        if(tr[p][!u])
        {
            p=tr[p][!u];
            ans+=(1<<i);
        }
        else p=tr[p][u];
    }
    return ans;
}

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

    while(scanf("%d",&n)!=EOF)
    {
        v.clear();
        memset(h,-1,sizeof (h)); idx=0;
        memset(tr,0,sizeof (tr)); tot=0;
        for(int i=1;i<=n-1;i++)
        {
            int a,b,c; scanf("%d%d%d",&a,&b,&c);
            add(a,b,c),add(b,a,c);
        }
        dfs(0,-1,0);
        int l=v.size(),ans=0;
        for(int i=0;i<l;i++)
        {
            int t=v[i];
            ans=max(ans,query(t));
        }
        printf("%d\n",ans);
    }




	return 0;
}
/*

*/