【码不停题3.15】被包围的区域

最新推荐文章于 2023-02-09 12:53:33 发布

芋圆萌萌哒

最新推荐文章于 2023-02-09 12:53:33 发布

阅读量284

点赞数

分类专栏：面试题文章标签：面试题

本文链接：https://blog.csdn.net/MmmTHU/article/details/88581190

版权

面试题专栏收录该内容

12 篇文章 0 订阅

订阅专栏

给定一个二维的矩阵，包含 ‘X’ 和 ‘O’（字母 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
解释:

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

class Solution {
public:
    int n, m;
    void solve(vector<vector<char>>& board) {
        n = board.size();
        if (!n) return;
        m = board[0].size();
        if (!m) return;
        
        for (int i = 0; i < n; i++) {
            if (board[i][0] == 'O') {
                dfs(board, i, 0);
            }
            if (board[i][m - 1] == 'O') {
                dfs(board, i, m - 1);
            }
        }
        for (int i = 0; i < m; i++) {
            if (board[0][i] == 'O') {
                dfs(board, 0, i);
            }
            if (board[n - 1][i] == 'O') {
                dfs(board, n - 1, i);
            }
        }
        for(int i = 0; i < n; ++i){
            for(int j = 0; j < m; ++j){
                if('-' == board[i][j])
                    board[i][j] = 'O';
                else if('O' == board[i][j])
                    board[i][j] = 'X';
            }
        }
        
    }
    void dfs(vector<vector<char>>& board, int i, int j) {
        board[i][j] = '-';
        if (i > 0 && board[i - 1][j] == 'O') {
            dfs(board, i - 1, j);
        }
        if (j > 0 && board[i][j - 1] == 'O') {
            dfs(board, i, j - 1);
        }
        if (i < n - 1 && board[i + 1][j] == 'O') {
            dfs(board, i + 1, j);
        }
        if (j < m - 1 && board[i][j + 1] == 'O') {
            dfs(board, i, j + 1);
        }
    }
};