算法学习 - 图的深度优先遍历(DFS) (C++)

时间:2021-09-02 10:13:10

深度优先遍历

在图的遍历中,其中深度优先遍历和广度优先遍历是最常见,也最简单的两种遍历方法。

深度优先遍历的思想就是一直向下找,找到尽头之后再去其他分支查找。

在上一篇博客中我已经写了广度优先遍历(BFS)。
想看的传送门:图的广度优先遍历

代码实现

这里实现和BFS的差别在于,在BFS中,我们使用的容器是队列(queue),是先进先出的, 而在DFS中我们需要使用的是栈(stack)一个先进后出的容器。

其他基本原理相同。

//
// main.cpp
// DFS
//
// Created by Alps on 15/4/1.
// Copyright (c) 2015年 chen. All rights reserved.
//

#include <iostream>
#include <stack>

#define Vertex int
#define WHITE 0
#define GRAY 1
#define BLACK 2

#define NumVertex 4

using namespace std;

struct node{
int val;
int weight; // if your graph is weight graph
struct node* next;
node(int v, int w):val(v), weight(w), next(NULL){}
};

typedef node* VList;

struct TableEntry{
VList header;
Vertex color;
};

typedef TableEntry Table[NumVertex+1];

void InitTableEntry(Vertex start, Table T){
Vertex outDegree;
VList temp = NULL;
for (int i = 1; i <= NumVertex; i++) {
scanf("%d", &outDegree);
T[i].header = NULL;
T[i].color = WHITE;
for (int j = 0; j < outDegree; j++) {
temp = (VList)malloc(sizeof(struct node));
scanf("%d %d", &temp->val, &temp->weight);
temp->next = T[i].header;
T[i].header = temp;
}
}
T[start].color = GRAY;
}

void DFS(Vertex start, Table T){
stack<Vertex> S;
S.push(start);
Vertex V;
VList temp;
while (!S.empty()) {
V = S.top();
printf("%d ", V);
T[V].color = BLACK;
S.pop();
temp = T[V].header;
while (temp) {
if (T[temp->val].color == WHITE) {
S.push(temp->val);
T[temp->val].color = GRAY;
}
temp = temp->next;
}
}
}

int main(int argc, const char * argv[]) {
Table T;

InitTableEntry(1, T);

DFS(1, T);

return 0;
}

测试用例:

2
2 1
4 1
0
1
2 1
2
2 1
3 1

这个是输入,每行的数字是节点出度,紧跟的是其相邻节点。

输出: 1 2 4 3

如此。此测试用例也可以用到我上一篇博客的BFS中。