牛客练习赛32B Xor Path （树形dp）

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

给定一棵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了。

备注:

题目大意：

给你一棵树，每个节点有一个权值。path[i,j]记录i到j的最短路径上所有节点权值的抑或。求所有path[i,j]的抑或（i=1~n-1,j=i+1~n）。注意(i,j)是有序实数对哦。

其实就是求每个点在所有最短路径中经历了几次。

树形dp就好了。假设1是树的根，sum[]记录子树的大小。dp过程还是有点妙哒。见代码。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <queue>
#include <deque>
#include <stack>
#include <map>
#include <set>
typedef long long ll;
const int mod=;
const int inf=;
const int maxn=;
const int maxm=;

int n;
int to[maxn*+];
int next[maxn*+];
int head[maxn+],cnt;
int val[maxn+];
ll times[maxn+];

int dfs(int x,int fa)
{
    int sum=;
    for(int i=head[x];i!=-;i=next[i])
    {
        int l=to[i];
        if(l!=fa)
        {
            int temp=dfs(l,x);
            times[x]+=(ll)sum*temp;
            sum+=temp;
        }
    }
    times[x]+=(ll)sum*(n--sum);
    times[x]+=n-;
    return sum+;
}

int main()
{
    scanf("%d",&n);
    memset(head,-,sizeof(head));
    cnt=;
    for(int i=,a,b;i<=n-;i++)
    {
        scanf("%d%d",&a,&b);
        to[cnt]=b;next[cnt]=head[a];head[a]=cnt++;
        to[cnt]=a;next[cnt]=head[b];head[b]=cnt++;
    }
    for(int i=;i<=n;i++)
        scanf("%d",val+i);

    memset(times,,sizeof(times));
    dfs(,-);

    int ans=^;
    for(int i=;i<=n;i++)
    {
        if(times[i]%==)
        {
            val[i]^=val[i];
        }
        ans^=val[i];
        //printf("%lld %d\n",times[i],ans);
    }

    printf("%d\n",ans);

    return ;
}