【dfs】surrounded regions

最新推荐文章于 2021-02-02 15:36:37 发布

Joseph_L_

最新推荐文章于 2021-02-02 15:36:37 发布

阅读量173

点赞数

CC 4.0 BY-SA版权

分类专栏：保研机试 LeetCode

本文链接：https://blog.csdn.net/m0_38033475/article/details/92073869

保研机试同时被 2 个专栏收录

85 篇文章

订阅专栏

LeetCode

54 篇文章

订阅专栏

本文介绍了一种解决特殊矩阵问题的算法，目标是在二维字符矩阵中，将未被X包围的O替换为X，但保留边界上的O不变。通过从边界上的'O'开始进行深度优先搜索并标记，再将剩余'O'变为X，最后将标记恢复为'O'，实现了高效解决问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个和油田问题不同的点在于，油田问题是求有多少块O，而这个还有个限制条件是：这块O必须被X包围。

我最开始的想法就是dfs里再套一个dfs，就是不进找到是O，还要先去判断一下和它同一块的O没在边界上，再去把这一块O变成X。但是果然时间太多了！所以，看题解看到一个很“简单粗暴”的方法：

从四个边上的'O'开始深度优先搜索将其标记为'a'
然后将剩下的'O'变成X
最后将'a'变回'O'

AC代码：

链接：https://www.nowcoder.com/questionTerminal/c159db5028034aa595043a1a220a62dd
来源：牛客网

class Solution {
public:
    void dfs(vector<vector<char>> &board, int row, int col,
            int x, int y){
        int stepx[4] = {1, -1, 0, 0}; //用数组保存路径
        int stepy[4] = {0, 0, 1, -1}; //吐槽一下C++这种在类里面的题目没法用全局变量
        if(board[x][y] == 'O'){
            board[x][y] = 'a';
            for(int i = 0; i < 4; i++){  //上下左右四个方向搜索
                if(x + stepx[i] >= 0 &&  x + stepx[i] < row &&
                   y + stepy[i] >= 0 && y + stepy[i] < col &&
                  board[x + stepx[i]][y + stepy[i]] == 'O'){
                    dfs(board, row, col, x + stepx[i], y + stepy[i]);
                }
            }
        }
    }
    void solve(vector<vector<char>> &board) {
        int row = board.size(); if(row == 0) return; //防止出很贱的测试样例
        int col = board[0].size();
        for(int i = 0; i < col; i++) dfs(board, row, col, 0, i);  //第一行
        for(int i = 0; i < col; i++) dfs(board, row, col, row - 1, i);  //最后一行
        for(int i = 0; i < row; i++) dfs(board, row, col, i, 0);  //第一列
        for(int i = 0; i < row; i++) dfs(board, row, col, i, col - 1);  //最后一列
        for(int i = 0; i < row; i++)
            for(int j = 0; j < col; j++){
                if(board[i][j] == 'O') board[i][j] = 'X';
                if(board[i][j] == 'a') board[i][j] = 'O';
            }
    }
};