北邮机试 | bupt oj | 100. 二叉树的层数 | 二叉树层次遍历

二叉树的层数

时间限制 1000 ms 内存限制 65536 KB

题目描述

老师有一个问题想考考mabo，但是mabo不会，所以想请你来帮帮忙。

问题如下：

给一个二叉树

请把这个棵二叉树按层来打印。如果为相同层，需要从左到右打印。一个节点是先添加左节点后添加右节点，即添加顺序与输入顺序一致。

输入格式

首先输入一个整数T，表示一共有T组数据 0<T<=10

再输入两个整数N,M(0<=N,M<=100)

表示下面有N行，这个树有M个节点(1号节点是这棵树的根节点)

每一行两个整数a,b(1<=a,b<=M)

表示节点a的父亲是节点b

输出格式

对于每组

先输出一行 "Qi:"表示第i个问题

然后接下来输出每个问题二叉树每层的节点，在同一层的节点用空格分开，同一层输出在一行（每一行末尾没有空格），不同的层输出在不同行（入下面Sample Ouput所示）

输入样例

2
4 5
2 1
3 1
4 2
5 4
1 2
2 1

输出样例

Q1:
1
2 3
4
5
Q2:
1
2

AC代码

#include<bits/stdc++.h>
#define MAXN 110
using namespace std;
struct node{
    vector<int > child;
}BNode[110];
void initTree(int line,int cnt){
    for(int i=1;i<=cnt;i++){
        BNode[i].child.clear();
    }
    int son,father;
    while(line--){
        scanf("%d%d",&son,&father);
        BNode[father].child.push_back(son);
    }
}
void levelOrder(int root,int num){
    printf("Q%d:\n",num);
    queue<int > q;
    q.push(root);
    while(!q.empty()){
        int size_q = q.size();
        while(size_q--){
            int front_q = q.front();
            q.pop();
            printf("%d",front_q);
            if(size_q){
                printf(" ");
            }
            else{
                printf("\n");
            }
            vector<int>& v = BNode[front_q].child;
            for(unsigned int i=0;i<v.size();i++){
                q.push(v[i]);
            }
        }
    }
}
int main()
{
    int t,n,m;
    scanf("%d",&t);
    for(int i=1;i<=t;i++){
        scanf("%d%d",&n,&m);
        initTree(n,m);
        levelOrder(1,i);
    }
}