很神的一道题。参考PoPoQQQ的题解:
首先转化为以
1
为根的有根树,
那么有
b1=∑ai×disi
bi=bfai−sizei+(size1−sizei)
并且有
ai=sizei−∑sizesoni
根据以上式子,我们有了 sizei(i>1) 和 size1 的关系,和 ai 和 sizei 的关系,所以就可以得到 ai 和 size1 的关系,又因为 b1=∑ai×disi ,所以可以得到 b1 和 size1 的关系,就得到了 size1 ,然后就能求出所有的 ai 了。
#include<iostream>
#include<cstdio>
using namespace std;
const int N=300005;
int b[N],head[N],list[N<<1],next[N<<1];
double dis[N],ans[N],m;
int n,cnt;
struct W
{
double a,b;
W (double x=0,double y=0) {a=x,b=y;}
W operator+(W x) {return W(a+x.a,b+x.b);}
W operator-(W x) {return W(a-x.a,b-x.b);}
W operator*(double x) {return W(a*x,b*x);}
};
W size[N],a[N],now;
inline int read()
{
int a=0,f=1; char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
return a*f;
}
inline void insert(int x,int y)
{
next[++cnt]=head[x];
head[x]=cnt;
list[cnt]=y;
}
inline void dfs(int x,int f)
{
size[x].a=0.5; size[x].b=0.5*(b[f]-b[x]);
a[x]=size[x];
for (int i=head[x];i;i=next[i])
if (list[i]!=f)
{
dis[list[i]]=dis[x]+1;
dfs(list[i],x);
a[x]=a[x]-size[list[i]];
}
}
int main()
{
n=read();
for (int i=1;i<n;i++)
{
int u=read(),v=read();
insert(u,v); insert(v,u);
}
for (int i=1;i<=n;i++) b[i]=read();
dfs(1,0);
for (int i=2;i<=n;i++)
now=now+(a[i]*dis[i]);
ans[1]=m=((double)(b[1]-now.b))/now.a;
for (int i=2;i<=n;i++)
ans[1]-=(ans[i]=a[i].a*m+a[i].b);
for (int i=1;i<=n;i++)
printf("%lld%c",(long long)ans[i],i==n?'\n':' ');
return 0;
}