hdu 5416 CRB and Tree 求树上路径异或结果为s的有多少

题意：给一颗树，定义f(u,v)为u到v路径上的异或结果，现在给定s，要求异或结果为s的路径有多少条。(1,2)和(2,1)只算一种。

根据异或的特性有f(u,v)=f(1,u)^f(1,v)，先把f(1,i)的所有i都求出，然后对于每一个u(1~n)如果有f(u,v)=s的话那么有f(1,v)=s^f(1,u)，所以求一下f(1,v)有多少个，加起来就行了，最后要特殊讨论下s==0的情况。

//author: CHC//First Edit Time:    2015-08-22 15:41#include <iostream>#include <cstdio>#include <cstring>#include <cmath>#include <set>#include <vector>#include <map>#include <queue>#include <set>#include <algorithm>#include <limits>using namespace std;typedef long long LL;const int MAXN=100000*2+1000;const int INF = numeric_limits<int>::max();const LL LL_INF= numeric_limits<LL>::max();struct Edge {    int to,next,w;    Edge(){}    Edge(int _to,int _next,int _w):to(_to),next(_next),w(_w){}}e[MAXN<<1];int head[MAXN],tot;void init(){    memset(head,-1,sizeof(head));    tot=0;}void AddEdge(int u,int v,int w){    e[tot]=Edge(v,head[u],w);    head[u]=tot++;    e[tot]=Edge(u,head[v],w);    head[v]=tot++;}int dis[MAXN];int has[1<<17],N,Q,T;void dfs(int u,int fa){    ++has[dis[u]];    for(int i=head[u];~i;i=e[i].next){        int v=e[i].to;        if(v==fa)continue;        if(dis[v]!=-1)continue;        dis[v]=dis[u]^e[i].w;        dfs(v,u);    }}int main(){    scanf("%d",&T);    while(T--){        scanf("%d",&N);        init();        for(int i=0,x,y,w;i<N-1;i++){            scanf("%d%d%d",&x,&y,&w);            AddEdge(x,y,w);        }        memset(has,0,sizeof(has));        memset(dis,-1,sizeof(dis));        dis[1]=0;        dfs(1,-1);        scanf("%d",&Q);        LL ans=0;        int s;        while(Q--){            scanf("%d",&s);            ans=0;            for(int i=1;i<=N;i++)ans+=has[s^dis[i]];            if(!s)ans+=N;            ans/=2;            printf("%I64d\n",ans);        }    }    return 0;}