使用栈的递归与非递归DFS

转载请注明来自 b0t0w1’blog

看到许多神人的DFS算法模板,大多是根据具体题目或者使用了 各种 struct 甚至是面向对象，不利于入门者学习。鄙人根据别人写的的非递归使用stack的DFS，写了份使用递归的模板，顺便分享那份非递归DFS模板。只是单纯遍历。读者可以自己添加具体功能。

- 递归_栈的DFS

#include <stdio.h> 
#include <memory.h>
#include <iostream>
#include <stack>
using namespace std;

//最大结点数
#define SIZE 100
//初始化距离无穷大
#define MAX 0x3f3f3f3f

//使用STL的stack，可以百度一下
stack<int> s;
//储存边的信息
int map[SIZE][SIZE];
//结点数
int N;
//记录遍历
int vis[SIZE];

void dfs(){
    //遍历栈顶元素并弹栈
    int ctop = s.top(); 
    s.pop();
    printf("%d ",ctop);
    //将与 ctop 距离为1的点压入栈中
    for (int i = 1;i <= N;i++){
        if(!vis[i] && map[i][ctop] == 1){
            vis[i] = 1;
            s.push(i);
        }
    }
    /*
    如果栈非空就继续调用dfs(),否则则退出。
    栈空的情况是：
        上面弹栈是已弹空，然后for循环未找到未遍历的点
    */
    if (!s.empty()){
        dfs();
    }
}

int main()
{
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
    int a,b,M;
    //读入点数边数
    scanf("%d%d",&N, &M);
    //初始化 map[]和vis[]
    memset(map,MAX,SIZE * SIZE);
    memset(vis,0,SIZE);
    //读入边的信息
    for (int i = 0; i < M; i++)
    {
        scanf("%d%d",&a,&b);
        //两点相连，默认为1
        map[b][a] = map[a][b] = 1;
    }
    //清空 栈s
    while(!s.empty()){
        s.pop();
    }
    //将点1 压入栈，默认从点1开始遍历。
    s.push(1);
    vis[1] = 1;
    dfs();
    return 0;
}

输入：

输出：

• 附：别人写的非递归，使用栈的dfs()

#include <stdio.h> 
#include <memory.h>
#include <iostream>
#include <stack>
using namespace std;

#define max_node 20
#define max 20000

stack<int> s;
int map[max_node][max_node];
int n;
int vis[max_node];


void dfs(int start){
    int ctop;
    vis[start] = 1;
    printf("%d ",start);
    for (int i = 1;i <= n;i++)
        if(!vis[i] && map[i][start] == 1)
        {
            vis[i] = 1;
            s.push(i);
        }
    while(!s.empty())
    {
        ctop = s.top();
        vis[ctop] = 1;
        printf("%d ",s.top());
        s.pop();
        for (int i = 1;i <= n;i++)
            if(!vis[i] && map[i][ctop] == 1)
            {
                vis[i] = 1;
                s.push(i);
            }
    }
}

int main()
{
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
    int a,b,m;
    scanf("%d%d",&n, &m);
    memset(map,max,400);
    memset(vis,0,max_node);
    for (int i = 0; i < m; i++)
    {
        scanf("%d%d",&a,&b);
        map[b][a] = map[a][b] = 1;
    }
    while(!s.empty()){
        s.pop();
    }
    dfs(1);
    return 0;
}

做了些小修改，可以直接使用上面的数据。结果相同。

转载请注明来自 b0t0w1’blog