【北邮OJ】100. 二叉树的层数

题目描述

老师有一个问题想考考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

学不来大佬的sao操作，只能借用队列暴力求解，中规中矩的思路

#include <cstdio>
#include <queue>
using namespace std;
queue <int> q; 
struct node{
	int count;//记录孩子节点个数 
	int child[3]; 
}Node[101];

int main()
{
	int t;
	scanf("%d",&t);
	int k=1;
	while(t--)
    {

    	for(int i=0;i<101;i++)
    	{
    	    Node[i].count=0;
    	}
    	int n,m;
    	scanf("%d%d",&n,&m);
    	for(int i=0;i<n;i++)
    	{
    		int a,b;
    		scanf("%d%d",&a,&b);
    		Node[b].child[Node[b].count]=a;
    		Node[b].count++;    		
    	}
    	while(!q.empty()) q.pop();
    	q.push(1);
    	int temp=1;//记录本层所有节点的个数 
    	int num=0;//记录正在遍历到本层的第num个节点 
    	int sum=0;//记录本层子节点的个数，作为下一层所有节点个数使用 
            	printf("Q%d:\n",k);
    	k++;
    	while(!q.empty())
    	{
    		int c=q.front();
    		q.pop();
    		printf("%d",c);
    		for(int i=0;i<Node[c].count;i++)
    		{
    			int ch=Node[c].child[i];
    			q.push(ch);
    			sum++;
    		}
    		num++;
    		if(num==temp) //本层遍历结束 
			{
				printf("\n");
				temp=sum;
				num=0;
				sum=0;
			}else {
				printf(" ");
			}
    	}
    }
}