leetcode, LC21: surrounded-regions包围区域

1 题目描述

现在有一个仅包含‘X’和‘O’的二维板，请捕获所有的被‘X’包围的区域
捕获一个被包围区域的方法是将被包围区域中的所有‘O’变成‘X’
例如

X X X X
X O O X
X X O X
X O X X

执行完你给出的函数以后，这个二维板应该变成：

X X X X
X X X X
X X X X
X O X X

2 解题思路

分析题目后，我们发现我们只需要保留其中有边缘在四条边上含有 O 的区域即可。

1. 遍历4条边上的元素，如果该元素为 O ，则将其改为 t（for temp） ，并深度优先搜索其周围的元素是否也为 O ，若是，则也改为 t 。
2. 经过第一步之后，在四条边上含有 O 的区域变为了 t ，内部含有 O 的区域仍为 O ，此时我们先遍历整张二维板将所有的内部 O 变为 X，再遍历整张二维板第二遍让所有的 t 变为 O，即可实现题目要求。

3 代码实现

class Solution {
public:
    void solve(vector<vector<char>> &board) {
        if(board.empty()) return;
        
        numOfRow = board.size();
        numOfCol = board[0].size();
        for(int i = 0; i < numOfCol; i++){
            dfs(board, 0, i);
            dfs(board, numOfRow - 1, i);
        }
        for(int i = 0; i < numOfRow; i++){
            dfs(board, i, 0);
            dfs(board, i, numOfCol - 1);
        }
        for(int i = 0; i < numOfRow; i++)
            for(int j = 0; j < numOfCol; j++)
                if(board[i][j] == 'O')
                    board[i][j] = 'X';
        for(int i = 0; i < numOfRow; i++)
            for(int j = 0; j < numOfCol; j++)
                if(board[i][j] == 't')
                    board[i][j] = 'O';
    }
    
private:
    int numOfRow = 0;
    int numOfCol = 0;
    
    void dfs(vector<vector<char>> &board, int row, int col){
        if(board[row][col] == 'O'){
            board[row][col] = 't';
            if(row > 0)
                dfs(board, row - 1, col);
            if(row < numOfRow - 1)
                dfs(board, row + 1, col);
            if(col > 0)
                dfs(board, row, col - 1);
            if(col < numOfCol - 1)
                dfs(board, row, col + 1);
        }
    }
};

4 运行结果

运行时间：5ms
占用内存：692k