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