LeetCode 130. 被围绕的区域（Surrounded Regions）

题目描述

给定一个二维的矩阵，包含 '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'。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

解题思路

分为两步：

• 首先对矩阵四条边界上的所有'O'进行深度优先搜索，并将所有与边界相连区域的'O'填充为'#'
• 然后遍历整个矩阵，将当前所有'O'，即处在被 'X' 围绕的区域的'O'填充为'X'，并将上一步填充的所有'#'还原为'O'

代码

 class Solution {
 public:
     void solve(vector<vector<char>>& board) {
         if(board.empty() || board[].empty()) return;
         int rows = board.size(), cols = board[].size();
         for(int i = ; i < cols; i++){
             if(board[][i] == 'O') fillBorder(board, , i);
             if(board[rows - ][i] == 'O') fillBorder(board, rows - , i);
         }
         for(int i = ; i < rows; i++){
             if(board[i][] == 'O') fillBorder(board, i, );
             if(board[i][cols - ] == 'O') fillBorder(board, i, cols - );
         }
         for(int i = ; i < rows; i++){
             for(int j = ; j < cols; j++){
                 if(board[i][j] == 'O') board[i][j] = 'X';
                 if(board[i][j] == '#') board[i][j] = 'O';
             }
         }
     }
     void fillBorder(vector<vector<char>>& board, int row, int col){
         board[row][col] = '#';
         if(row -  >=  && board[row - ][col] == 'O') fillBorder(board, row - , col);
         if(row +  < board.size() && board[row + ][col] == 'O') fillBorder(board, row + , col);
         if(col -  >=  && board[row][col - ] == 'O') fillBorder(board, row, col - );
         if(col +  < board[].size() && board[row][col + ] == 'O') fillBorder(board, row, col + );
     }
 };