Leetcode 130. 被围绕的区域【BFS】

问题描述

给定一个二维的矩阵,包含 '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' 换成其他字母,例如 'Q'
  • 之后 BFS 搜索,当搜到符合条件的 'O' 时,压入队列中,将其置为 'Q'
  • BFS 结束后,将矩阵中的 'Q' 置为 'O',其他置为 'X'

这题还可以用 并查集 实现,待补充。

扫描二维码关注公众号,回复: 11377401 查看本文章

实现代码

class Solution {
public:
    unordered_map<int,int>parent;
    unordered_map<int,int>cnt;
    void solve(vector<vector<char>>& board) {
        if(board.size()==0) return;
        int m=board.size(), n=board[0].size();
        int dx[4]={0,1,0,-1};
        int dy[4]={1,0,-1,0};
        queue<pair<int,int>>q;
        
        for(int i=0;i<m;i++){
            if(board[i][0]=='O'){
                q.push({i,0});
                board[i][0]='Q';
            }
            if(board[i][n-1]=='O'){
                q.push({i,n-1});
                board[i][n-1]='Q';
            }
        }
        for(int j=0;j<n;j++){
            if(board[0][j]=='O'){
                q.push({0,j});
                board[0][j]='Q';
            }
            if(board[m-1][j]=='O'){
                q.push({m-1,j});
                board[m-1][j]='Q';
            }
        }
        // cout<<q.size()<<endl;
        while(!q.empty()){
            pair<int, int>one=q.front();
            q.pop();
            for(int i=0;i<4;i++){
                int x=one.first+dx[i];
                int y=one.second+dy[i];
                if(x>=0&&x<m&&y>=0&&y<n&&board[x][y]=='O'){
                    q.push({x,y});
                    board[x][y]='Q';
                }
            }
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(board[i][j]=='Q') board[i][j]='O';
                else board[i][j]='X';
            }
        }
    }
};

参考资料

[1] Leetcode 130. 被围绕的区域

猜你喜欢

转载自blog.csdn.net/qq_27690765/article/details/106616399