跳转至

DS图遍历--深度优先搜索

时间限制: 1 Sec 内存限制: 128 MB

题目描述

给出一个图的邻接矩阵，对图进行深度优先搜索，从顶点0开始

注意：图n个顶点编号从0到n-1

代码框架如下：

DS-1223-1.jpg

DS-1223-2.jpg

输入

第一行输入t，表示有t个测试实例

第二行输入n，表示第1个图有n个结点

第三行起，每行输入邻接矩阵的一行，以此类推输入n行

第i个结点与其他结点如果相连则为1，无连接则为0，数据之间用空格隔开

以此类推输入下一个示例

输出

每行输出一个图的深度优先搜索结果，结点编号之间用空格隔开

样例输入

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
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
2
0 2 1 3 
0 3 2 1 4

提示

解决方案

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include <iostream>
#include <vector>

class Map {
public:
    Map(int size) : size(size), matrix(size, std::vector<int>(size, 0)) {}
    void setMatrixFromCin() {
        for (int ir = 0; ir < matrix.size(); ++ir) {
            for (int ic = 0; ic < matrix.size(); ++ic) {
                std::cin >> matrix[ir][ic];
            }
        }
    }
    void dfs() {
        visited = std::vector<bool>(size, false);
        for (int i = 0; i < visited.size(); ++i) {
            if (visited[i] == false) {
                dfs(i);
            }
        }
        std::cout << std::endl;
    }

private:
    int size;
    std::vector<std::vector<int> > matrix;
    std::vector<bool> visited;
    void dfs(int vertex) {
        visited[vertex] = true;
        std::cout << vertex << ' ';
        for (int i = 0; i < matrix.size(); ++i) {
            if (matrix[vertex][i] == 1 && visited[i] == false) {
                dfs(i);
            }
        }
    }
};

int main() {
    int ctrl;
    std::cin >> ctrl;

    while (ctrl--) {
        int size;
        std::cin >> size;
        Map map(size);
        map.setMatrixFromCin();
        map.dfs();
    }

    return 0;
}

评论