第一个思路是,从点阵里面的O出发,考察这些O是否被X包围,若是,则将这些O变成X. 但判定这些O是否被X包围比较困难,我没有做成;
然后看了别人的题解,逆向思维,从边缘的O出发(上下左右四排的O),进行DFS, 将这些'O'可达的所有点'O'都标记为'Z'(包括它们自身也标记为O), 这些'Z'最终都可以保持为'O'不变,而其他点应该都变成'X'. 这样就可以了。
但注意到,如果直接DFS, 在数据量比较大时,会由于递归的栈深度过深,出现栈溢出(提示Runtime Error).
所以不同与网上的部分题解,应该用stack<Position>去保存这些点,模拟递归。
代码:
struct Position
{
Position(int x, int y) : x(x), y(y) {}
int x, y;
};
class Solution
{
public:
void solve(vector<vector<char>> &board)
{
if (board.size()==0 || board[0].size()==0)
{
return ;
}
int n = board.size();
int m = board[0].size();
vector<vector<bool>> vi(board.size(), vector<bool>(board[0].size(), false));
stack<Position> s;
for (int i = 0; i < n; ++ i)
{
s.push(Position(i, 0));
s.push(Position(i, m - 1));
}
for (int j = 0; j < m; ++ j)
{
s.push(Position(0, j));
s.push(Position(n - 1, j));
}
while (s.empty() == false)
{
Position p = s.top();
s.pop();
if (p.x>=0 && p.x<n && p.y>=0 && p.y<m)
{
if (vi[p.x][p.y]==false && board[p.x][p.y]=='O')
{
vi[p.x][p.y] = true;
board[p.x][p.y] = 'Z';
s.push(Position(p.x - 1, p.y));
s.push(Position(p.x + 1, p.y));
s.push(Position(p.x, p.y - 1));
s.push(Position(p.x, p.y + 1));
}
}
}
for (int i = 0; i < n; ++ i)
{
for (int j = 0; j < m; ++ j)
{
board[i][j] = (board[i][j]=='Z')? 'O': 'X';
}
}
}
};