题目链接
题目描述:
给定一个二维的矩阵,包含 ‘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'
。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
解题思路
要仔细读题,题目实际上可以简化为一句话:找出边界O的连通区域。说的直白些就是判断边界是否存在O,如果存在,则以这个点O向四个方向扩散,其所在方向上的O不能改变,直到遇见X或者到了另一边界为止。
一开始,我想一劳永逸直接在双循环遍历中朝四个方向找O,忽略了这四个方向的O即使属于本结点的内部结点(也就是遍历结点四个方向边界都是X),然而却不一定属于它自己方向的内部结点,结果倒在了测试用例31。
后来,经过题解区大神们的启发,可以先找到边界为O的位置,然后向四个方向DFS寻找O的连通区域,将该区域值置为F(任意取X和O之外的值即可),之后剩下的O就肯定是内部结点了,最后双循环还原即可。
代码
class Solution {
public:
void solve(vector<vector<char>>& board) {
if(board.size() <= 2)
return;
int row = board.size(), line = board[0].size();
for(int i = 0; i < row; i++){
for(int j = 0; j < line; j++){
bool flag = i == 0 || i == row - 1 || j == 0 || j == line - 1;
if(flag && board[i][j] == 'O')
helper(board, i, j);
}
}
for(int i = 0; i < row; i++){
for(int j = 0; j < line; j++){
if(board[i][j] == 'O')
board[i][j] = 'X';
else if(board[i][j] == 'F')
board[i][j] = 'O';
}
}
return;
}
private:
void helper(vector<vector<char>>& board, int i, int j){
if(i < 0 || i >= board.size() || j < 0 || j >= board[0].size() || board[i][j] == 'X' || board[i][j] == 'F')
return;
board[i][j] = 'F';
helper(board, i - 1, j);
helper(board, i + 1, j);
helper(board, i, j - 1);
helper(board, i, j + 1);
return;
}
};
如果有错误或者不严谨的地方,请务必给予指正,十分感谢。