题面传送门
稍微看一下就可以得出结论:一个人获得的硬币数就是以这个点为根子树内所有节点的深度和。
那么树形
d
p
dp
dp就可以直接处理。
f
x
=
(
∑
(
u
,
x
)
∈
E
f
u
+
s
i
z
u
)
+
1
f_x=(\sum\limits_{(u,x)∈E}{f_u}+siz_u)+1
fx=((u,x)∈E∑fu+sizu)+1
代码实现:
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,k,x,y,siz[200039];
long long f[200039];
struct yyy{int to,z;};
struct ljb{
int head,h[200039];
yyy f[200039];
inline void add(int x,int y){
f[++head]=(yyy){y,h[x]};
h[x]=head;
}
}s;
inline void dfs(int x){
siz[x]=1;
int cur=s.h[x];
yyy tmp;
while(cur!=-1){
tmp=s.f[cur];
dfs(tmp.to);
f[x]+=f[tmp.to];
siz[x]+=siz[tmp.to];
cur=tmp.z;
}
f[x]+=siz[x];
}
int main(){
memset(s.h,-1,sizeof(s.h));
register int i;
scanf("%d",&n);
for(i=2;i<=n;i++) scanf("%d",&x),s.add(x,i);
dfs(1);
for(i=1;i<=n;i++) printf("%lld ",f[i]);
}