B 网2——二叉树的层数
Accept:34 | Submit:366 |
Time Limit:1000MS | Memory Limit:65536KB |
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 <stdio.h>
#include <queue>
using namespace std;
struct btnode
{
int data ;
int level;
btnode *lnode;
btnode *rnode;
};
void merge(btnode &a , btnode &b)
{
if(b.lnode==NULL)
{
b.lnode = &a;
}
else
{
b.rnode = &a;
}
a.level = b.level+1;
}
int main()
{
int t;
int n , m ;
int i ,j ;
int a , b;
queue<btnode> Q;
btnode bt[100];
scanf("%d", &t);
for(i = 0 ; i<t ; i++)
{
scanf("%d%d", &n ,&m);
for(j = 1 ; j<=m ; j++)
{
bt[j].level = 0;
bt[j].lnode = NULL;
bt[j].rnode = NULL;
}
bt[1].level = 1;
for(j = 0 ; j<n ; j++)
{
scanf("%d%d", &a , &b);
bt[a].data = a ;
bt[b].data = b ;
merge(bt[a],bt[b]);
}
/* for(j = 1 ; j<=m ; j++)
{
printf("btnodelevel%d : %d " , j ,bt[j].level);
}
printf("\n");
*/
Q.push(bt[1]);
int flag = 0 ;
btnode b ;
while(!Q.empty())
{
b = Q.front();
Q.pop();
if(b.level!=flag)
{
if(b.level==1)
printf("%d", b.data);
else
printf("\n%d", b.data);
flag = b.level;
}
else
{
printf(" %d" ,b.data);
}
if(b.lnode!=NULL)
Q.push(*b.lnode);
if(b.rnode!=NULL)
Q.push(*b.rnode);
}
printf("\n);
}
return 0;
}