题目内容:

給一棵tree,還有他的root,輸出這棵tree的level order traversal。
Note : 一個節點可能有大於兩個的孩子節點。

Note : 當一個節點有多個孩子需要造訪時,由數字編號小的節點開始造訪。



输入格式:
第一行為一個正整數T(T<100),代表測資數,接下來每筆測資如下格式:
第一行有2個整數,N R,分別代表N個點以及root為編號R的節點,所有點的編號為1~N。
  接下來有N-1行,每行有2個數字u v,代表編號u的節點與編號v的節點相連。
數字範圍:
0 < N <= 1000


输出格式:
每棵tree輸出一行,level order traversal的結果,數字間以空格隔開。


输入样例:
2
3 1
1 2
1 3
5 5
1 2
4 5
3 5
5 2 


输出样例:
1 2 3
5 2 3 4 1


时间限制:1000ms内存限制:128000kb

解法:

将图用邻接矩阵表示,使用队列进行广度优先搜索。


代码:

#include <iostream>

#include <queue>
/* run this program using the console pauser or add your own getch, system("pause") or input loop */


void getResult(int num, int root){
	int a,b,i,j;
	int k,graph[num+1][num+1], explored[num+1];
	for(k=0;k<num+1;k++){
		for(j=0;j<num+1;j++){
			graph[k][j] = 0;
		}
	}
	for(k=0;k<num+1;k++){
		explored[k] = 0;
	}
	// input num-1 relationship
	for(i=1;i<num;i++){
		std::cin>>a>>b;
		graph[a][b] = 1;
		graph[b][a] = 1;
	}
	int explored_num = 0, value;
	std::queue<int> que;
	que.push(root);
	while(explored_num<num){
		value = que.front();
		que.pop();
		if(explored[value]==0){
			if(explored_num == num-1){
				std::cout<<value;
			}
			else{
				std::cout<<value<<" ";
			}
			explored_num++;
			explored[value] = 1;
			for(i=1;i<=num;i++){
				if(graph[value][i]==1 && explored[i]==0){
					que.push(i);
				}
			}
		}
	}
}

int main() {
	int test_num, num, root;
	std::cin>>test_num;
	for(int i=0;i<test_num;i++){
		std::cin>>num>>root;
		getResult(num,root);
		if(i<test_num-1){
		    std::cout<<"\n";
	    }
	}
	return 0;
}