树的遍历 | 1004 广义树的层序遍历

最新推荐文章于 2022-05-06 09:12:16 发布

weixin_30302609

最新推荐文章于 2022-05-06 09:12:16 发布

阅读量139

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/TQCAI/p/8550996.html

本文分享了一道曾经让作者困扰不已的PAT竞赛题目，如今轻松解决的心得体会，并附上了详细的AC代码实现。该题涉及图的遍历算法，通过广度优先搜索统计每个层级的叶子节点数。

刚接触pat的时候就做过这个题。当时被难得要死，连题都看不懂，一脸蒙蔽。

时过境迁，今天一看这题就笑了，太简单了。对着电脑一顿操作，就AC了。

AC代码：

#include <stdio.h>
#include <memory.h>
#include <math.h>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <map>

#define I scanf
#define OL puts
#define O printf
#define F(a,b,c) for(a=b;a<c;a++)
#define FF(a,b) for(a=0;a<b;a++)
#define FG(a,b) for(a=b-1;a>=0;a--)
#define LEN 1010
#define MAX (1<<30)-1
#define V vector<int>

using namespace std;

typedef struct Node{
    int id;
    vector<int> l;    //leaves
}Node;

Node nd[LEN];

int main(){
//    freopen("1004.txt","r",stdin);
    int n,m,k,t,i,id;
    I("%d%d",&n,&m);
    while(m--){
        I("%d%d",&id,&k);
        while(k--){
            I("%d",&t);
            nd[id].l.push_back(t);
        }
    }
    queue<int> q;
    q.push(1);
    vector<int> ans;
    while(!q.empty()){
        int sz=q.size();
        int cnt=0;
        while(sz--){
            t=q.front();
            q.pop();
            if(nd[t].l.size()){
                FF(i,nd[t].l.size()) q.push(nd[t].l[i]);
            }else
                cnt++;    
        }
        ans.push_back(cnt);
    }
    FF(i,ans.size()){
        O("%d",ans[i]);
        if(i!=ans.size()-1) O(" ");
    }
    return 0;
}