栈实现DFS算法

最新推荐文章于 2024-10-25 09:41:30 发布

D.Liang-TechBlog

最新推荐文章于 2024-10-25 09:41:30 发布

阅读量2.1k

点赞数

分类专栏：数据结构与算法 C++ 文章标签：数据结构

本文链接：https://blog.csdn.net/qq_38844835/article/details/126111721

版权

C++ 同时被 2 个专栏收录

26 篇文章 0 订阅

订阅专栏

数据结构与算法

9 篇文章 2 订阅

订阅专栏

该博客介绍了如何利用栈来实现DFS算法，以解决递归深度过深导致的爆栈问题。在二维网格中，算法遍历1表示的岛屿，计算其面积并更新最大面积，过程中将已访问节点压入栈中，并逐个检查相邻节点。这种方法有效地优化了传统递归DFS的性能。

摘要由CSDN通过智能技术生成

DFS是搜索算法中非常常见的一种算法，通常实现的时候会通过递归调用的方式来实现，但是众所周知，当递归深度很深时可能会出现爆栈的情况。

针对上述可能出现的问题，这里通过栈来实现DFS操作。

int dims[5] = { -1, 0, 1, 0, -1 }; //四个方向的实现。

stack<DIM> stack_dfs; //定义栈结构，用于存储每次DFS的节点。

int grid_data[100][100];
int max_area = 0;
struct DIM {
    int x;
    int y;
};
// 定义坐标
int dims[5] = { -1, 0, 1, 0, -1 };
stack<DIM> stack_dfs;
void DFS(int row, int col) {
    if (grid_data[row][col] == 1) {
        int island_area = 1;
        DIM datas = { row, col };
        stack_dfs.emplace(datas);
        grid_data[row][col] = 0;
        while (!stack_dfs.empty()) {
            row = stack_dfs.top().x;
            col = stack_dfs.top().y;
            stack_dfs.pop();
            for (int i = 0; i < 4; i++) {
                int x = row + dims[i];
                int y = col + dims[i + 1];
                if (isValide(x, y) && grid_data[x][y] == 1) {
                    DIM datas = { x, y };
                    stack_dfs.emplace(datas);
                    grid_data[x][y] = 0;
                    ++island_area;
                }
            }
        }
        max_area = max(island_area, max_area);
    }
}

// 用于计算最大值，尽量使用内联函数，提高效率
inline int max(const int& a, const int& b) {
    if (a > b) return a;
    return b;
}

// 判断坐标是否有效
inline bool isValide(const int& x, const int& y) {
    if (x >= 0 && x < max_x && y <= 0 && y < max_y) {
        return true;
    }
    return false;
}