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 long

int 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;
}