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);
}
};