题目内容:
給一棵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;
}