leetcode 130. 被围绕的区域 medium

leetcode 130. 被围绕的区域  medium          

题目描述：

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

解题思路：

1. 遍历矩阵，在4个边界处，假如当前元素是O，那就执行dfs，把遍历到的O都给换成 #

2. 再次遍历矩阵，这次把所有O都给换成X，而把所有#都给换成O

代码：

class Solution {
public:
    void solve(vector<vector<char>>& board) {
        if(board.empty())
            return;
        int row=board.size();
        int col=board[0].size();
        
        //遍历矩阵，在4个边界处，假如当前元素是O，那就执行dfs，把遍历到的O都给换成 #
        for(int i=0;i<row;++i)
            for(int j=0;j<col;++j){
                if((i==0 || i==row-1 || j==0 || j==col-1) && board[i][j]=='O')
                    dfs(i,j,board);
            }
        
        
        // 再次遍历矩阵，这次把所有O都给换成X，而把所有#都给换成O
        for(int i=0;i<row;++i)
            for(int j=0;j<col;++j){
                if(board[i][j]=='O')
                    board[i][j]='X';
                else if(board[i][j]=='#')
                    board[i][j]='O';
            }
        
        return;
        
    }
    
    
    void dfs(int i,int j,vector<vector<char>> &board){
        if(i<0 || j<0 || i>=board.size() || j>=board[0].size() || board[i][j] != 'O')
            return;
        board[i][j]='#';
        dfs(i+1,j,board);
        dfs(i-1,j,board);
        dfs(i,j+1,board);
        dfs(i,j-1,board);
    }
    
};