二叉树的层数

题目描述:

老师有一个问题想考考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:"表示第1个问题，然后接下来输出每个问题二叉树每层的节点，在同一层的节点用空格分开，同一层输出在一行(每一行末尾没有空格) ,不同的层输出在不同行(下面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

#include<bits/stdc++.h>
using namespace std;
int main() {
	int T;
	int N,M,a,b;
	int level[105];
	int max_level;
	cin>>T;
	int k=1;//Qi 
	while(T--) {
		max_level=1;
		memset(level,0,sizeof(level));
		cin>>N>>M;
		while(N--) {
			level[1]=1;//根节点的层数记为1
			cin>>a>>b;//a是b的子节点
			level[a]=level[b]+1;
			if(level[a]>max_level) {
				max_level=level[a];
			}
		}
		cout<<"Q"<<k<<":"<<endl;
		k++; 
		int count;//空格计数 
		for(int i=1; i<=max_level; i++) {
			count=0;
			for(int j=1; j<=M; j++) {
				if(level[j]==i){
					count++;
					if(count==1){
						cout<<j;
					}
					else cout<<" "<<j;
				}
			}
			cout<<endl;
		}
	}
}