【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);}