E. 广度优先搜索-STL对象版

题目描述

广度优先搜索-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;
}