子树大小 自底向上累加
2025-02-04 09:53:29
发布于:上海
#include<bits/stdc++.h>
using namespace std;
vector<int> tr[10010];
int n;
int ans[10010];
//返回以u为根的子树的大小传给父节点
int dfs(int u){
int size_u = 1;
for(int son:tr[u])size_u+=dfs(son);
ans[u] = size_u;
return size_u;
}
int main(){
cin>>n;
for(int i=2;i<=n;i++){
int fa;
cin>>fa;
tr[fa].push_back(i);
}
dfs(1);
for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
return 0;
}
这里空空如也
有帮助,赞一个