BZOJ 3727 PA2014 树DP 解题报告

3727: PA2014 Final Zadanie

Description

吉丽YY了一道神题，题面是这样的：“一棵n个点的树，每条边长度为1，第i个结点居住着a[i]个人。假设在i结点举行会议，所有人都从原住址沿着最短路径来到i结点，行走的总路程为b[i]。输出所有b[i]”
吉丽已经造好了数据，但熊孩子把输入文件中所有a[i]给删掉了。你能帮他恢复吗？

Input

第一行一个整数n(2<=n<=300000)。
接下来n-1行，每行两个整数x,y，表示x和y之间有连边。
接下来一行由空格隔开的n个整数bi。

Output

输出一行由空格隔开的n个整数a[i]。
如果你觉得有多组解就任意输出其中一组。

Sample Input

2
1 2
17 31

Sample Output

31 17

【解题报告】
做不来，看了http://blog.csdn.net/CreationAugust/article/details/50178921 的题解

/**************************************************************    Problem: 3727    User: onepointo    Language: C++    Result: Accepted    Time:3492 ms    Memory:42936 kb****************************************************************/#include<cstdio>#include<cstring>#include<algorithm>using namespace std;#define N 300010#define LL long longint n,cnt,head[N],deep[N];struct Edge{int to,nxt;}e[N<<1];LL ans,suma,sumb;LL size1[N],size2[N],a1[N],a2[N],b[N];void adde(int u,int v){    e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;    e[++cnt].to=u;e[cnt].nxt=head[v];head[v]=cnt;}void dfs(int u,int fa){    for(int i=head[u];~i;i=e[i].nxt)    {        int v=e[i].to;        if(v==fa) continue;        size2[v]=size1[v]=b[u]-b[v];        a1[v]=a2[v]=1;        deep[v]=deep[u]+1;        dfs(v,u);        size1[u]-=size2[v];        a1[u]-=a2[v];    }    suma+=size1[u]*deep[u];    sumb+=a1[u]*deep[u];}int main(){    cnt=-1;    memset(head,-1,sizeof(head));    scanf("%d",&n);    for(int i=1;i<n;++i)    {        int u,v;scanf("%d%d",&u,&v);        adde(u,v);    }    for(int i=1;i<=n;++i) scanf("%d",&b[i]);    a1[1]=2;dfs(1,1);    ans=(b[1]*2-suma)/sumb;    for(int i=1;i<=n;++i)          printf("%lld%c",(a1[i]*ans+size1[i])>>1,i==n?'\n':' ');    return 0;}