Description
给出一个图的邻接矩阵,对图进行广度优先搜索,从顶点0开始
注意:图n个顶点编号从0到n-1
如果图不连通,则对尚未访问的编号结点继续进行广度优先搜索,直到所有结点被访问
Input
第一行输入t,表示有t个测试实例
第二行输入n,表示第1个图有n个结点
第三行起,每行输入邻接矩阵的一行,以此类推输入n行
第i个结点与其他结点如果相连则为1,无连接则为0,数据之间用空格隔开
以此类推输入下一个示例
Output
每行输出一个图的广度优先搜索结果,结点编号之间用空格隔开
Sample
Input
2 4 0 0 1 1 0 0 1 1 1 1 0 1 1 1 1 0 5 0 0 0 1 1 0 0 1 0 0 0 1 0 1 1 1 0 1 0 0 1 0 1 0 0Output
0 2 3 1 0 3 4 2 1
此方法只适用于连通图,因为oj的前台后台数据都是连通图所以可以过。
思路:
直接处理邻接矩阵,从第0行开始,遍历每一行的全部数据,若为1则存入队列,并且将这一列的所有值变成0。这说明已经遍历过这个节点了,从下一个节点开始遍历行的时候,就表明这一节点已经访问过。然后从队头开始递归,一个res队列存访问顺序,temp队列存遍历数据,不断进行入队和出队处理。当temp队列空了,依次输出res队列即可。
#include <iostream>
#include <queue>
using namespace std;
queue<int> res;
queue<int> temp;
int k = 0;
int p = 0;
void DFS(int** arr, int n, int v) {
if (!temp.empty())
//若队列不为空则队头出队
temp.pop();
for (int j = 0; j < n; j++) {//对每一行进行遍历
if (arr[v][j] == 1) {
//当前这一列都置为0
for (int k = 0; k < n; k++) {
arr[k][j] = 0;
}
arr[j][v] = 0;//两个节点相互连接,例如v1到v3已经访问了,就不用再看v3到v1。请不要复制粘贴
temp.push(j);
res.push(j);
}
}
if (!temp.empty())
//如果栈不为空,对队头数据进行递归。
DFS(arr, n, temp.front());
else {
//如果栈空了则输出结果
while (!res.empty()) {
cout << res.front() << " ";
res.pop();
}
}
}
int main() {
int t;
cin >> t;
while (t--) {
int n;
cin >> n;
int** arr = new int* [n];
for (int i = 0; i < n; i++){
arr[i] = new int[n];
for (int j = 0; j < n; j++) {
cin >> arr[i][j];
}
}
cout << "0 ";
DFS(arr, n, 0);
cout << endl;
}
return 0;
}