Level Order Traversal 图的层序遍历 cpp

题目内容：

給一棵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;
}