Description
老师有一个问题想考考mabo,但是mabo不会,所以想请你来帮帮忙。
问题如下:
给一个二叉树
请把这个棵二叉树按层来打印。如果为相同层,需要从左到右打印。一个节点是先添加左节点后添加右节点,即添加顺序与输入顺序一致。
Input
首先输入一个整数T,表示一共有T组数据 0<T<=10
再输入两个整数N,M(0<=N,M<=100)
表示下面有N行,这个树有M个节点(1号节点是这棵树的根节点)
每一行两个整数a,b(1<=a,b<=M)
表示节点a的父亲是节点b
Output
对于每组
先输出一行 "Qi:"表示第i个问题
然后接下来输出每个问题二叉树每层的节点,在同一层的节点用空格分开,同一层输出在一行(每一行末尾没有空格),不同的层输出在不同行(入下面Sample Ouput所示)
SampleInput
2
4 5
2 1
3 1
4 2
5 4
1 2
2 1
SampleOutput
Q1:
1
2 3
4
5
Q2:
1
2
#include <bits/stdc++.h>
using namespace std;
struct node
{
int x;
int l, r;
int dept;
node()
{
l = r = 1;
}
};
int main()
{
int t;
scanf("%d", &t);
for(int k = 1; k <= t; k++)
{
int n, m;
scanf("%d%d", &n, &m);
node s[1100];
s[1].l = s[1].r = -1;
s[1].dept = 1;
int mdept = 1;
for(int i = 1; i <= n; i++)
{
int a, b;
scanf("%d%d", &a, &b);
if(s[b].l == -1)
s[b].l = a;
else
s[b].r = a;
s[a].dept = s[b].dept+1;
mdept = max(mdept, s[a].dept);
}
// cout<<mdept<<endl;
printf("Q%d:\n",k);
int cnt = 1;
node tmp = s[1];
while(cnt <= mdept)
{
int flag = 0;
for(int i = 1; i <= m; i++)
{
if(flag && s[i].dept == cnt)
{
printf(" %d",i);
}
else if(!flag&&s[i].dept == cnt)
{
printf("%d",i);
flag=1;
}
}
printf("\n");
cnt++;
}
}
return 0;
}