1 题目描述
现在有一个仅包含‘X’和‘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
2 解题思路
分析题目后,我们发现我们只需要保留其中有边缘在四条边上含有 O 的区域即可。
- 遍历4条边上的元素,如果该元素为 O ,则将其改为 t(for temp) ,并深度优先搜索其周围的元素是否也为 O ,若是,则也改为 t 。
- 经过第一步之后,在四条边上含有 O 的区域变为了 t ,内部含有 O 的区域仍为 O ,此时我们先遍历整张二维板将所有的内部 O 变为 X,再遍历整张二维板第二遍让所有的 t 变为 O,即可实现题目要求。
3 代码实现
class Solution {
public:
void solve(vector<vector<char>> &board) {
if(board.empty()) return;
numOfRow = board.size();
numOfCol = board[0].size();
for(int i = 0; i < numOfCol; i++){
dfs(board, 0, i);
dfs(board, numOfRow - 1, i);
}
for(int i = 0; i < numOfRow; i++){
dfs(board, i, 0);
dfs(board, i, numOfCol - 1);
}
for(int i = 0; i < numOfRow; i++)
for(int j = 0; j < numOfCol; j++)
if(board[i][j] == 'O')
board[i][j] = 'X';
for(int i = 0; i < numOfRow; i++)
for(int j = 0; j < numOfCol; j++)
if(board[i][j] == 't')
board[i][j] = 'O';
}
private:
int numOfRow = 0;
int numOfCol = 0;
void dfs(vector<vector<char>> &board, int row, int col){
if(board[row][col] == 'O'){
board[row][col] = 't';
if(row > 0)
dfs(board, row - 1, col);
if(row < numOfRow - 1)
dfs(board, row + 1, col);
if(col > 0)
dfs(board, row, col - 1);
if(col < numOfCol - 1)
dfs(board, row, col + 1);
}
}
};
4 运行结果
运行时间:5ms
占用内存:692k