祺哥刷题日记（三） PTA甲级 1004 Counting Leaves

最新推荐文章于 2021-03-23 12:57:27 发布

Vo Racci

最新推荐文章于 2021-03-23 12:57:27 发布

阅读量148

点赞数

分类专栏：刷题

本文链接：https://blog.youkuaiyun.com/weixin_43849763/article/details/109732602

版权

刷题专栏收录该内容

5 篇文章

订阅专栏

本文介绍如何使用深度优先搜索（DFS）算法解决输入家谱信息的问题，计算每代中没有后代的人数。通过构建树结构并遍历，找出各代的孤独节点数量。

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

题目

在这里插入图片描述
题目很长，大致的意思就是输入家谱信息，输出每代人中没有后代的人的个数。

思路

看到题目中第一行的tree，就可以很自然地想到使用树来存储家谱，相关的算法也不过就是与树相关的算法。
找到每一代没有后代的人数，首先要知道每个人的代数，也就是在家谱树中所在的层数，所以可以使用dfs算法解决问题。

代码

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

typedef struct people{
    int level = 0;
    vector<int> children;
}people;
vector<people> family(100);    //存储家谱
vector<int> leaves(100);
int maxLevel = 0;

int transID2Int(string ID){
    return (ID[0] - '0') * 10 + (ID[1] - '0') - 1;
}

void dfs(int p, int depth){
    if(family[p].children.size() == 0){
        maxLevel = max(depth,maxLevel);
        leaves[depth] ++;
        return;
    }
    else{
        family[p].level += depth;
        depth ++;
        for(int i = 0; i < family[p].children.size(); i++){
            dfs(family[p].children[i],depth);
        }
    }
}
int main(){
    int n, m, k;
    string parent, child;
    cin >> n >> m;
    //建立家谱
    for(int i = 0; i < m; i++){
        cin >> parent >> k;
        for(int j = 0; j < k; j++){
            cin >> child;
            int childNum = transID2Int(child);
            family[transID2Int(parent)].children.push_back(childNum);
        }
    }
    dfs(0,0);
    
    cout<<leaves[0];
    for(int i=1;i<=maxLevel;i++){
        cout<<" "<<leaves[i];
    }
    
    return 0;
}