北邮12年网研 -二叉树的层数

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;
}