Leetcode 130：被围绕的区域

题目描述

给定一个二维的矩阵，包含 '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:
    vector<int> dx = {-1,0,1,0};
    vector<int> dy = {0,-1,0,1};
    void dfs(vector<vector<char>>& board,int m,int n,int i,int j){
        board[i][j] = 'S';
        for(int k=0;k<4;++k){
            int ti = i+dx[k];
            int tj = j+dy[k];
            if(ti>=0&&ti<m&&tj>=0&&tj<n&&board[ti][tj]=='O') dfs(board,m,n,ti,tj);
        }
    }
    void solve(vector<vector<char>>& board) {
        if(board.size() == 0 || board[0].size() == 0) return;
        int m = board.size(),n = board[0].size();
        for(int i=0;i<m;++i){
            if(board[i][0] == 'O') dfs(board,m,n,i,0);
            if(board[i][n-1] == 'O') dfs(board,m,n,i,n-1);
        }
        for(int j=0;j<n;++j){
            if(board[0][j] == 'O') dfs(board,m,n,0,j);
            if(board[m-1][j] == 'O') dfs(board,m,n,m-1,j);
        }
        for(int i=0;i<m;++i){
            for(int j=0;j<n;++j){
                auto &c = board[i][j];
                if(c == 'O') c='X';
                if(c == 'S') c='O';
            }
        }
    }
};