9-3 C. DS图遍历--深度优先搜索

题目描述

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

代码框架参考课本P169算法7.4和7.5，同学们可在理解的基础上自行设计算法，不强制要求完全相同

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

输入

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

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

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

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

以此类推输入下一个示例

输入样例：

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

输出

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

输出样例：

0 2 1 3 
0 3 2 1 4 

代码

#include <iostream>
using namespace std;

class List{
public:
    int n;
    int **matrix;  //矩阵
    int *flag;  //标志是否输出
    List(){
        cin >> n;
        matrix = new int*[n];
        for(int i = 0; i < n; i++){  //输入矩阵数据
            matrix[i] = new int[n];
            for(int j = 0; j < n; j++){
                cin >> matrix[i][j];
            }
        }

        flag = new int[n];  //初始化
        for(int i = 0; i < n; i++){
            flag[i] = 0;  // 0 还未输出
        }
    }

    void DFS(){
        int num=0,key=0;  //num标志已经输出的顶点个数，key标志当前处理的顶点编号
        while(num < n){
            cout << key << " ";
            flag[key] = 1;
            num++;
            int flag1=0;  //表示本趟是否找到输出顶点 0未找到
            for(int i = 0; i < n; i++){
                if(matrix[key][i] == 1 && flag [i] != 1){
                    key = i;
                    flag1 = 1;
                    break;
                }
            }
            if(flag1 == 0){  //若未找到，则遍历flag,找到优先出现的未输出的顶点编号
                for(int i = 0; i < n; i++){
                    if(flag[i] == 0){
                        key = i;
                        break;
                    }
                }
            }
        }
        cout << endl;
    }
};

int main()
{
    int t;
    cin >> t;
    while(t--){
        List list;
        list.DFS();
    }
    return 0;
}