C++LeetCode 每日一题 -被围绕的区域(补）

题目链接:https://leetcode-cn.com/problems/surrounded-regions/

Dscription

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

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

Sample

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

PS

被围绕的区间不会存在于边界上，换句话说，任何边界上的 ‘O’ 都不会被填充为 ‘X’。 任何不在边界上，或不与边界上的 ‘O’ 相连的 ‘O’ 最终都会被填充为 ‘X’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

Solution

显然任何边界上的 'O’都不会被填充为 ‘X’，那我们只需要边界上的 'O’都标记一下，然后将其他的 'O’填充为 ‘X’。直接无脑BFS！

AC Code

class Solution {
public:
    int dx[4] = { 1, -1, 0, 0 };
    int dy[4] = { 0, 0, 1, -1 };
    queue< pair<int,int> > s;
    void solve(vector<vector<char> >& board) {
        int n = board.size();
        if(n<=2) return ;
        int m = board[0].size();
        if(m<=2) return ;
        
        for (int i = 0; i < m; i++) {
            if (board[0][i] == 'O') s.push({ 0, i });
            if (board[n - 1][i] == 'O') s.push({ n - 1, i });
        }
        for (int i = 1; i < n - 1; i++) {
            if (board[i][0] == 'O') s.push({ i, 0 });
            if (board[i][m - 1] == 'O') s.push({ i, m - 1 });
        }
        while (!s.empty()) {
            int x = s.front().first,y = s.front().second;
            board[x][y] = '#';
            s.pop();
            for (int i = 0; i < 4; i++) {
                int mx = x + dx[i], my = y + dy[i];
                 if (mx < 0 || my < 0 || mx >= n || my >= m || board[mx][my] != 'O') {
                    continue;
                }
                s.push({mx,my});
            }
        }
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (board[i][j] == '#') board[i][j] = 'O';
                else if (board[i][j] == 'O') board[i][j] = 'X';
            }
        }
    }
};

觉得有帮助的话，点个赞再走吧！