LeetCode - 130. 被围绕的区域

130. 被围绕的区域

给定一个二维的矩阵，包含 ‘X’ 和 ‘O’（字母 O）。

找到所有被 ‘X’ 围绕的区域，并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。

解题思路： 从边上的’O’开始DFS，将搜索到的‘O’换个记号，搜索完毕后，剩余的‘O’即为需要用‘X’填充的‘O’。

class Solution {
public:
    void helper(vector<vector<char>>& board, int r, int c) {
        int rows = board.size(), cols = board[0].size();
        if (r < 0 || r >= rows || c < 0 || c >= cols || board[r][c] != 'O') return;
        board[r][c] = '$';
        helper(board, r + 1, c);
        helper(board, r - 1, c);
        helper(board, r, c + 1);
        helper(board, r, c - 1);
    }
    void solve(vector<vector<char>>& board) {
        if (board.empty() || board[0].empty()) return ;
        int rows = board.size(), cols = board[0].size();
        for (int j = 0; j < cols; ++j) {
            helper(board, 0, j);
            helper(board, rows - 1, j);
        }
        for (int i = 1; i < rows - 1; ++i) {
            helper(board, i, 0);
            helper(board, i, cols - 1);
        }
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if (board[i][j] == 'O') board[i][j] = 'X';
                if (board[i][j] == '$') board[i][j] = 'O';
            }
        }
    }
};