D. Decorate Apple Tree （树的dfs）（*1600）

最新推荐文章于 2023-04-30 16:29:52 发布

GocNeverGiveUp

最新推荐文章于 2023-04-30 16:29:52 发布

阅读量698

点赞数

CC 4.0 BY-SA版权

分类专栏： Codeforces 简单搜索

本文链接：https://blog.youkuaiyun.com/qq_41444888/article/details/84591216

Codeforces 同时被 2 个专栏收录

38 篇文章

订阅专栏

简单搜索

20 篇文章

订阅专栏

本文介绍了一种使用vector构建二叉树的方法，并通过深度优先搜索（DFS）算法递归计算每个节点下包含的叶子节点数量。代码示例清晰展示了从输入构建树结构到最终输出排序后的叶子节点计数的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://codeforces.com/contest/1056/problem/D

问题：求每个节点包含多少个叶子（叶子本身包含自身）

知识点：运用vector建立一颗二叉树，然后通过dfs递归的方式回推每个节点包含多少叶子，算是简单的vector建立关系的应用

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
#define ll long long
#define mod 1000000007
#define inf 0x3f3f3f3f
using namespace std;
int num[100005];
vector<int>v[100005];
void dfs(int x)
{
    if(! v[x].size())
    {
        num[x] = 1;
        return ;
    }
    for(int i = 0; i < v[x].size(); i ++)
    {
        dfs(v[x][i]);
        num[x] += num[v[x][i]];
    }
    return ;
}
int main()
{
    int n;
    scanf("%d",&n);
    int tmp;
    for(int i = 2; i <= n; i ++)
    {
        scanf("%d",&tmp);
        v[tmp].push_back(i);
    }
    dfs(1);
    sort(num + 1, num + 1 + n);
    for(int i = 1; i < n; i ++)
        printf("%d ",num[i]);
    printf("%d\n",num[n]);
    return 0;
}