被围绕的区域

给定一个二维的矩阵,包含 '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',都替换成'Z',然后将二维矩阵中的所有'O'替换成'X',将所有'Z'还原替换成'O'
class Solution {
public:
    void solve(vector<vector<char>>& board) {
        if (board.empty())
            return;
        
        if (board[0].empty())
            return;
        
        for(int i = 0; i < board.size(); ++i)
            if(board[i][0] == 'O')
                bfs(board, i, 0);
        
        for(int i = 0; i < board.size(); ++i)
            if(board[i][board[0].size() - 1] == 'O')
                bfs(board, i, board[0].size() - 1);
        
        for(int i = 0; i < board[0].size(); ++i)
            if(board[0][i] == 'O')
                bfs(board, 0, i);
        
        
        for(int i = 0; i < board[0].size(); ++i)
            if(board[board.size() - 1][i] == 'O')
                bfs(board, board.size() - 1, i);
        
        for(int i = 0; i < board.size(); ++i){
            for(int j = 0; j < board[0].size(); ++j){
                if('Z' == board[i][j])
                    board[i][j] = 'O';
                else if('O' == board[i][j])
                    board[i][j] = 'X';
            }
        }
        
    }
    
    void bfs(vector<vector<char>>& board, int i, int j){
        board[i][j] = 'Z';
        if (i > 0 && board[i - 1][j] == 'O')
            bfs(board, i - 1, j);
        if (i < board.size() - 1 && board[i + 1][j] == 'O')
            bfs(board, i + 1, j);
        if (j > 0 && board[i][j - 1] == 'O')
            bfs(board, i, j - 1);
        if (j < board[0].size() - 1 && board[i][j + 1] == 'O')
            bfs(board, i, j + 1);
    } 
};

猜你喜欢

转载自blog.csdn.net/leaf_scar/article/details/88554231