牛客练习赛32 B题 Xor Path

链接：https://ac.nowcoder.com/acm/contest/272/B
来源：牛客网

题目描述

给定一棵n个点的树，每个点有权值。定义表示  到  的最短路径上，所有点的点权异或和。

对于，求所有的异或和。

输入描述:

第一行一个整数n。

接下来n-1行，每行2个整数u,v，表示u,v之间有一条边。

第n+1行有n个整数，表示每个点的权值。

输出描述:

输出一个整数，表示所有

的异或和，其中

。

输入例子:

4
1 2
1 3
1 4
1 2 3 4

输出例子:

5

-->

示例1

输入

复制

4
1 2
1 3
1 4
1 2 3 4

输出

复制

5

说明

再将这6个数异或起来就可以得到答案5了。

备注:

 

题解：
　　考虑每个点的经过次数sum=子树经过该点到另一个子树+子树上的点经过该点到外面的点+该点到其他点(就是n-1);
如果sum为奇数则对答案有贡献，偶数没有贡献；最后将是奇数的点的权值异或即可；
参考代码：

 #include<bits/stdc++.h>
 using namespace std;
 #define clr(a,val) memset(a,val,sizeof(a))
 #define RI register int
 #define eps 1e-6
 typedef long long ll;
 const int INF=0x3f3f3f3f;
 inline ll read()
 {
     ll x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-') f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }
 const int maxn=1e6+;
 ll n,u,v,tot,head[maxn],val[maxn];
 ll cnt[maxn],ans;
 struct Edge{
     ll u,v,nxt;
 } edge[maxn];
 inline void Init() {clr(head,-);ans=;tot=;clr(cnt,);}
 inline void addedge(ll u,ll v)
 {
     edge[tot].u=u;
     edge[tot].v=v;
     edge[tot].nxt=head[u];
     head[u]=tot++;
 }
 inline void dfs(ll t,ll pre)
 {
     cnt[t]=; ll sum=;
     for(ll i=head[t];~i;i=edge[i].nxt)
     {
         ll v=edge[i].v;
         if(v!=pre)
         {
             dfs(v,t);
             sum+=cnt[v]*(cnt[t]-);
             cnt[t]+=cnt[v];
         }
     }
     sum+=(cnt[t]-)*(n-cnt[t])+n-;
     if(sum&) ans^=val[t];
 }
 int main()
 {
     n=read(); Init();
     for(ll i=;i<n-;++i) { u=read(),v=read();addedge(u,v);addedge(v,u);}
     for(ll i=;i<=n;++i) val[i]=read();
     dfs(,);
     printf("%lld\n",ans);
     return ;
 }