问题描述
给定一个二维的矩阵,包含 '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'
。
这题还可以用 并查集 实现,待补充。
实现代码
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';
}
}
}
};