【dfs】surrounded regions

这个和油田问题不同的点在于,油田问题是求有多少块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';
            }
    }
};

猜你喜欢

转载自blog.csdn.net/m0_38033475/article/details/92073869