PAT A1004

clipboard.png

还是数层数和数节点的问题,个人觉得用BFS比较好;
当然用DFS也能做,具体的思路就是建立层数数组,深度遍历到x层的时候,如果是叶子节点就leaf[x]++,如果不是就不管,继续跳过;
代码片段可以这样:

void DFS(int index,int h){
    max_h=max(h,max_h);
    if(G[index].size()==0){
        leaf[h]++;
        return;
    }
    for(int i=0;i<G[index].size();i++){
        DFS(G[index][i],h+1);
    }
}

BFS代码如下所示:

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<vector>
#include<queue>
using namespace std;
using std::vector;
using std::queue;
const int maxn=110;
int n,m;
int leaf[maxn]={0};//每层leaf个数
vector<int> table[maxn];
int BFS(int x){
    int layer=0;
    queue<int>q;
    q.push(x);
    while(!q.empty()){
        int num=0;
        int length=q.size();
        layer++;
        for(int i=0;i<length;i++){
            int id=q.front();
            q.pop();
            if(table[id].size()==0){
                num++;
            }
            for(int j=0;j<table[id].size();j++){
                q.push(table[id][j]);
            }
        }
        leaf[layer]=num;
    }
    return layer;
}
int main(){
    int root,l,num;
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        scanf("%d %d",&root,&num);
        for(int j=0;j<num;j++){
            scanf("%d",&l);
            table[root].push_back(l);
        }
    }
    int layer=BFS(1);
    for(int i=1;i<=layer;i++){
        if(i==1)
            printf("%d",leaf[i]);
        else
            printf(" %d",leaf[i]);
    }
    system("pause");
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值