图的应用之——图的连通

Description

给定一个图的邻接矩阵，请判断该图是否是连通图。连通图：任意两个顶点之间都有路径。
–程序要求–
若使用C++只能include一个头文件iostream；若使用C语言只能include一个头文件stdio
程序中若include多过一个头文件，不看代码，作0分处理
不允许使用第三方对象或函数实现本题的要求

Input

第1行输入一个整数k，表示有k个测试数据
第2行输入一个整数n，表示有n个结点
从第3行起到第n+2行输入一个邻接矩阵，其中Matrix[i,j]=1表示第i，j个结点之间有边，否则不存在边。
接下来是第2到第k个测试数据的结点数和邻接矩阵

Output

输出Yes or No表示图是否是强连通图

Sample

Input
2
4
0 1 1 1
1 0 1 1
1 1 0 1
1 1 1 0
7
0 1 0 0 0 0 0
0 0 1 1 0 0 0
1 0 0 0 0 0 0
1 0 1 0 0 0 0
0 0 0 0 0 1 1
0 1 0 0 0 0 0
0 0 0 1 0 1 0
Output
Yes
No

思路：

dfs就是一条路走到黑，如果一点与其他的点联通，那么visited数组就会全部变成true

完整代码

#include<iostream>
using namespace std;
int** matrix;
bool* visited;
int n, flg;
void dfs(int pos) {
	visited[pos] = true;
	for (int i = 0; i < n; i++)
		if (!visited[i] && matrix[pos][i])
			dfs(i);
}
int main() {
	int t; cin >> t;
	while (t--) {
		cin >> n;
		matrix = new int* [n];
		visited = new bool[n];
		for (int i = 0; i < n; i++) {
			matrix[i] = new int[n];
			visited[i] = false;
			for (int j = 0; j < n; j++)
				cin >> matrix[i][j];
		}
		flg = 0;
		for (int i = 0; i < n; i++) {
			dfs(i);
			for (int j = 0; j < n; j++)
				if (!visited[j])
					flg = 1;
				else visited[j] = false;
			if (flg) break;
		}
		flg == 1 ? cout << "No" << endl : cout << "Yes" << endl;
	}
	return 0;
}