【PA2014】【BZOJ3727】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
HINT

Source

鸣谢Jcvb

如果没有Delayyy当时留下的题解根本不会做TAT果然人傻
另外Po姐写的更详细
PoPoQQQ

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 300010
#define LL long long
#define GET (ch>='0'&&ch<='9')
using namespace std;
int n,top;
int deep[MAXN];
LL size1[MAXN],size2[MAXN],a1[MAXN],a2[MAXN],ans,suma,sumb,b[MAXN];
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
struct edge
{
    int to;
    edge *next;
}e[MAXN<<1],*prev[MAXN];
void insert(int u,int v)    {e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top];}
void dfs(int x,int f)
{
    for (edge *i=prev[x];i;i=i->next)
        if (i->to!=f)
        {
            size2[i->to]=size1[i->to]=b[x]-b[i->to];a1[i->to]=a2[i->to]=1;
            deep[i->to]=deep[x]+1;dfs(i->to,x);
            size1[x]-=size2[i->to];a1[x]-=a2[i->to];
        }
    suma+=size1[x]*deep[x];sumb+=a1[x]*deep[x];
}
int main()
{
    in(n);int u,v,x;
    for (int i=1;i<n;i++)   in(u),in(v),insert(u,v),insert(v,u);
    for (int i=1;i<=n;i++)  in(x),b[i]=x;a1[1]=2;
    dfs(1,0);ans=(b[1]*2-suma)/sumb;
    for (int i=1;i<=n;i++)  printf(i!=n?"%lld ":"%lld\n",(a1[i]*ans+size1[i])>>1);
}