给定一个二维的矩阵,包含 ‘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)设置一个与board相同大小的矩阵flag,用来标识board中的真实值,先将flag全部置为0。
(2)我们需要注意到,只有当边界中存在O时,里面的区域才有可能不被置为X,因此我们要做的就是从边缘的O出发找到所有与之相连的区域并将其flag置为1。
(3)遍历内层board,将flag为0的区域全部置为X
C++代码:
class Solution {
public:
void solve(vector<vector<char>>& board) {
if (board.empty() || board[0].empty())
return;
int row = board.size();
int col = board[0].size();
vector<vector<int>> flag(board.size(), vector<int>(board[0].size(), 0));
// 设置标志位
int row_begin = 0, row_end = row - 1, col_begin = 0, col_end = col - 1;
// 第一行与最后一行
for (int j = 0; j < col; j++) {
if (board[row_begin][j] == 'O') {
dfs(board, flag, row_begin, j);
}
}
for (int j = 0; j < col; j++) {
if (board[row_end][j] == 'O') {
dfs(board, flag, row_end, j);
}
}
// 第一列与最后一列
for (int i = 1; i < row - 1; i++) {
if (board[i][col_begin] == 'O') {
dfs(board, flag, i, col_begin);
}
}
for (int i = 1; i < row - 1; i++) {
if (board[i][col_end] == 'O') {
dfs(board, flag, i, col_end);
}
}
// 修改board值
for (int i = 1; i < row - 1; i++) {
for (int j = 1; j < col - 1; j++) {
if (flag[i][j] == 0 && board[i][j] == 'O')
board[i][j] = 'X';
}
}
return;
}
void dfs(vector<vector<char>>& board, vector<vector<int>>& flag, int i, int j) {
if (0 <= i && i < board.size() && 0 <= j && j < board[0].size()) {
if (flag[i][j] == 1 || board[i][j] == 'X')
return;
flag[i][j] = 1;
dfs(board, flag, i + 1, j);
dfs(board, flag, i - 1, j);
dfs(board, flag, i, j + 1);
dfs(board, flag, i, j - 1);
}
}
};