题目描述
广度优先搜索-STL对象版
给出一个图的邻接矩阵,对图进行广度优先搜索,从顶点0开始
注意:图n个顶点编号从0到n-1
/
基本STL的用法:
1)
vector::push_back(1); //将1增加到数组尾部
vector::push_back(e); //将类型为E的一个对象e增加到数组尾部.其中类型E可以自行定义,如Vertex类,int类等。
2.1)
vector vec;
int x = vec[1]; //读取组数v(索引从0开始)的第1个元素
2.2)
vector v;
int& x = v[1]; //引用组数v的第1个元素,注意此处的引用&的用法
x = 10; //修改组数v的第1个元素为10
3.1)
queue my_queue;
my_queue.push(0); //将0增加到队列尾部
3.2)
int root = my_queue.front(); //读取队列的首元素
my_queue.pop(); //弹出队列的首元素
/
代码框架如下:
输入
第一行输入t,表示有t个测试实例
第二行输入n,表示第1个图有n个结点
第三行起,每行输入邻接矩阵的一行,以此类推输入n行
第i个结点与其他结点如果相连则为1,无连接则为0,数据之间用空格隔开
以此类推输入下一个示例
输出
每行输出一个图的广度优先搜索结果,结点编号之间用空格隔开
输入样例1
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 0
输出样例1
0 2 3 1
0 3 4 2 1
代码
#include<iostream>
#include<string>
#include<vector>
#include<queue>
using namespace std;
class Graph {
//邻接矩阵
vector<vector<int>> adjMat;
//顶点个数
int n;
//记录每个顶点的索引
vector<bool> isVisited;
public:
//测试输出
void printAdjMat() {
int i = 0;
for (; i < n; ++i) {
for (int j = 0; j < n; ++j)
cout << adjMat[i][j] << " ";
cout << endl;
}
}
//输入结点
void readAdjMat() {
cin >> this->n;
int i = 0;
for (; i < n; ++i) {
isVisited.push_back(false);
vector<int> row;
int j = 0;
for (; j < n; ++j) {
int edge;
cin >> edge;
row.push_back(edge);
}
adjMat.push_back(row);
}
//printAdjMat();
}
void update(queue<int>& my_queue,int rootNo) {
int j = 0;
for (; j < n; ++j) {
if (isVisited[j] == 1)
continue;
if (adjMat[rootNo][j] == 0)
continue;
isVisited[j] = 1;
my_queue.push(j);
}
}
void bfs(int startNo=0) {
isVisited[startNo] = 1;
queue<int> my_queue;
my_queue.push(startNo);
//bfs
while (!my_queue.empty()) {
int root;
root = my_queue.front();
my_queue.pop();
cout << root << " ";
update(my_queue, root);
}
cout << endl;
}
void main() {
readAdjMat();
bfs();
}
};
int main() {
int t;
cin >> t;
while (t--) {
Graph g;
g.main();
}
return 0;
}