1.题目描述
给定一个二维的矩阵,包含 ‘X’ 和 ‘O’(字母 O)。
找到所有被 ‘X’ 围绕的区域,并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。
示例:
运行你的函数后,矩阵变为:
解释:
被围绕的区间不会存在于边界上,换句话说,任何边界上的 ‘O’ 都不会被填充为 ‘X’。 任何不在边界上,或不与边界上的 ‘O’ 相连的 ‘O’ 最终都会被填充为 ‘X’。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
2.思路
任何与边界上的 O 相连的 O (包括边界上的O)都不会被填充为 X,因此对边界上的 O 进行深度优先遍历,把和其相连的 O 都标记为 A ,则未被标记的O就是被围绕的区域,再次遍历矩阵,把标记为 A 的元素设置为 O ,把 O设置为 X。
3.代码
class Solution {
public:
void dfs(vector<vector<char>>& board, int x, int y){
if(x < 0 || x >= n || y < 0 || y >= m || board[x][y] != 'O'){
return;
}
board[x][y] = 'A';
dfs(board, x + 1, y);
dfs(board, x - 1, y);
dfs(board, x, y + 1);
dfs(board, x, y - 1);
}
void solve(vector<vector<char>>& board) {
n = board.size();
if(n == 0){
return ;
}
m = board[0].size();
for(int i = 0;i < n;++i){
dfs(board, i, 0);
dfs(board, i, m - 1);
}
for(int i = 1;i < m-1;++i){
dfs(board, 0, i);
dfs(board, n - 1, i);
}
for(int i = 0;i < n;++i){
for(int j = 0;j < m;++j){
if(board[i][j] == 'A'){
board[i][j] = 'O';
}
else if(board[i][j] == 'O'){
board[i][j] = 'X';
}
}
}
}
private:
int n;
int m;
};
4.复杂度分析
时间复杂度:O(n×m),其中 n 和 m 分别为矩阵的行数和列数。深度优先搜索过程中,每一个点至多只会被标记一次。
空间复杂度:O(n×m),其中 n 和 m 分别为矩阵的行数和列数。主要为深度优先搜索的栈的开销。