被围绕的区域

题目链接

给你一个 m x n 的矩阵 board ，由若干字符 ‘X' 和 ‘O' ，找到所有被 ‘X' 围绕的区域，并将这些区域里所有的 ‘O' 用 ‘X' 填充

示例:

20220122185941-2022-01-22-18-59-42

输入：board = [["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"]]
解释：被围绕的区间不会存在于边界上，换句话说，任何边界上的 'O' 都不会被填充为 'X'。 任何不在边界上，或不与边界上的 'O' 相连的 'O' 最终都会被填充为 'X'。如果两个元素在水平或垂直方向相邻，则称它们是"相连"的。

思路

解法一：

题目要求任何边界上的 O 都不会被填充为 X， 所有的不被包围的 O 都直接或间接与边界上的 O 相连。

对于每一个边界上的 O，我们以它为起点，标记所有与它直接或间接相连的字母 O；

最后我们遍历这个矩阵，对于每一个字母：

  • 如果该字母被标记过，先将字符替换为 A， 该字母为没有被字母 X 包围的字母 O，后面将其还原为字母 O；
  • 如果该字母没有被标记过，则该字母为被字母 X 包围的字母 O，我们将其修改为字母 X。
class Solution {
public:
    int n, m;
    static constexpr int dirs[4][2] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};
    void solve(vector<vector<char>>& board) {
        n = board.size();
        if (n == 0) {
            return;
        }
        m = board[0].size();
        for (int i = 0; i < n; i++)
        {
            dfs(board, i, 0);
            dfs(board, i, m - 1);
        }

        for (int i = 1; i < m - 1; i++) {
            dfs(board, 0, i);
            dfs(board, n - 1, i);
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (board[i][j] == 'A') {// 被标记过，为没有被字母 X 包围的字母 O，我们将其还原为字母 O；
                    board[i][j] = 'O';
                }else if (board[i][j] == 'O') { 
                    board[i][j] = 'X'; //未被标记，为被字母 X 包围的字母 O，将 O 替换为 X
                }
            }
        }
    }

    void dfs(vector<vector<char>> &board, int x, int y) {
        if (x < 0 || y <0 || x >= n || y >= m || board[x][y] != 'O') {
            return;
        }
        board[x][y] = 'A'; // 将O替换为A

        for (int i = 0; i < 4; i++) {
            int nx = x + dirs[i][0];
            int ny = y + dirs[i][1];
            dfs(board, nx, ny);
        }
    }
};