原题地址:https://leetcode-cn.com/problems/surrounded-regions/description/
题目描述:
给定一个二维的矩阵,包含 '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'。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
解题方案:
本题用到了深度优先遍历和回溯法,解题方式也是之前未见过的,方法是很巧妙的。
首先想到的是对边界的'0'进行标记,与边界'0'相连的'0'也进行标记,并标记为'+';
其次查找连续'0'的方法,采用的是深度优先遍历,实现的过程回溯法。
最后没有被标记的'0'即为被包围的,将会被修改为'X',而'+'将被恢复为'0'。
class Solution {
public:
void dfs(vector<vector<char>>& board, int row, int col)
{
if(row < 0 || row >= board.size() || col < 0 || col >= board[0].size())
return;
if(board[row][col] != 'O')
return;
board[row][col] = '+';
dfs(board, row + 1, col);
dfs(board, row, col + 1);
dfs(board, row - 1, col);
dfs(board, row, col - 1);
}
void solve(vector<vector<char>>& board) {
if(board.size() <= 1 || board[0].size() <= 2)
return;
for(int i = 0; i < board.size(); i ++){
if(board[i][0] == 'O')
dfs(board, i, 0);
if(board[i][board[0].size() - 1] == 'O')
dfs(board, i, board[0].size() - 1);
}
for(int i = 0; i < board[0].size(); i ++){
if(board[0][i] == 'O')
dfs(board, 0, i);
if(board[board.size() - 1][i] == 'O')
dfs(board, board.size() - 1, i);
}
// 对'+'进行恢复,对'0'进行修改
for(int i = 0; i < board.size(); i ++){
for(int j = 0; j < board[0].size(); j ++){
if(board[i][j] == '+')
board[i][j] = 'O';
else if(board[i][j] == 'O')
board[i][j] = 'X';
}
}
}
};