这个和油田问题不同的点在于,油田问题是求有多少块O,而这个还有个限制条件是:这块O必须被X包围。
我最开始的想法就是dfs里再套一个dfs,就是不进找到是O,还要先去判断一下和它同一块的O没在边界上,再去把这一块O变成X。但是果然时间太多了!所以,看题解看到一个很“简单粗暴”的方法:
从四个边上的'O'开始深度优先搜索 将其标记为'a'
然后将剩下的'O'变成X
最后将'a'变回'O'
AC代码:
链接:https://www.nowcoder.com/questionTerminal/c159db5028034aa595043a1a220a62dd
来源:牛客网
class Solution {
public:
void dfs(vector<vector<char>> &board, int row, int col,
int x, int y){
int stepx[4] = {1, -1, 0, 0}; //用数组保存路径
int stepy[4] = {0, 0, 1, -1}; //吐槽一下C++这种在类里面的题目没法用全局变量
if(board[x][y] == 'O'){
board[x][y] = 'a';
for(int i = 0; i < 4; i++){ //上下左右四个方向搜索
if(x + stepx[i] >= 0 && x + stepx[i] < row &&
y + stepy[i] >= 0 && y + stepy[i] < col &&
board[x + stepx[i]][y + stepy[i]] == 'O'){
dfs(board, row, col, x + stepx[i], y + stepy[i]);
}
}
}
}
void solve(vector<vector<char>> &board) {
int row = board.size(); if(row == 0) return; //防止出很贱的测试样例
int col = board[0].size();
for(int i = 0; i < col; i++) dfs(board, row, col, 0, i); //第一行
for(int i = 0; i < col; i++) dfs(board, row, col, row - 1, i); //最后一行
for(int i = 0; i < row; i++) dfs(board, row, col, i, 0); //第一列
for(int i = 0; i < row; i++) dfs(board, row, col, i, col - 1); //最后一列
for(int i = 0; i < row; i++)
for(int j = 0; j < col; j++){
if(board[i][j] == 'O') board[i][j] = 'X';
if(board[i][j] == 'a') board[i][j] = 'O';
}
}
};