1004 Counting Leaves(30 分)【dfs/bfs】

6 篇文章 0 订阅
4 篇文章 0 订阅

题意:给你一个家谱树,找到没有孩子的人,即寻找每层的叶子结点数

方法一:用dfs。采用vector数组作为树的存储结构,将树结点逐一输入到vector 中,每一个元素存储了该结点的子孩子信息,自动向下探索子孩子,直到vector[i].size()==0表明该结点的子孩子为零,即叶子结点。 

#include <bits/stdc++.h>
using namespace std;

int N,M;
vector<int> v[101];
int res[101],maxdepth=-1;

void dfs(int f,int depth)
{
    if(v[f].size() == 0){
        res[depth]++;
        if(depth>maxdepth)
            maxdepth = depth;
        return;
    }
    for(int i=0;i<v[f].size();i++){
        dfs(v[f][i],depth+1);
    }
}

int main()
{
    memset(res,0,sizeof(int));
    cin>>N>>M;
    for(int i=0;i<M;i++){
        int father;
        cin>>father;
        int k;
        cin>>k;
        for(int j=0;j<k;j++){
            int child;
            cin>>child;
            v[father].push_back(child);
        }
    }
    dfs(1,0);
    cout<<res[0];
    for(int i=1;i<=maxdepth;i++)
        cout<<" "<<res[i];
    cout<<endl;

    return 0;
}

方法二:bfs

vector<int> v[101];
int result[101], level[101], maxLevel = -1; //level用来记录每个节点对应的层数
void bfs()
{
  queue<int> q;
  q.push(1);
  level[1] = 0;
  while(!q.empty())
  {
    int id = q.front();
    q.pop();
    if(maxLevel < level[id])
      maxLevel = level[id];
    if(v[id].size() == 0)
      result[level[id]]++;
    for(int i = 0;i < v[id].size();i++)
    {
      q.push(v[id][i]);
      level[v[id][i]] = level[id]+1;
    }
  }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值