给定一个二维的矩阵,包含 ‘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'。
如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。
逆向考虑问题,
我们先统计出哪些区域不会被攻占,然后将其它区域都变成'X'即可。
具体做法如下:
开一个二维布尔数组,记录哪些区域被遍历过。
枚举所有边界上的’O’,从该位置做Flood Fill,即做深度优先遍历,只遍历是’O’的位置,并将所有遍历到的位置都标记成true。
将所有未遍历到的位置变成’X’。
时间复杂度分析:每个位置仅被遍历一次,所以时间复杂度是 O(n2)O(n2),其中 nn 是地图的边长。
class Solution {
public:
vector<vector<bool>> st;
int n, m;
void solve(vector<vector<char>>& board) {
if(board.empty() || board[0].empty()) return;
n = board.size(), m = board[0].size();
st = vector<vector<bool>> (n,vector<bool>(m, false));
for(int i=0;i<n;i++){
if(board[i][0] == 'O') dfs(i,0,board);
if(board[i][m-1] == 'O') dfs(i,m-1,board);
}
for(int i=0;i<m;i++){
if(board[0][i] == 'O') dfs(0,i,board);
if(board[n-1][i] == 'O') dfs(n-1,i,board);
}
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
if(!st[i][j])
board[i][j] = 'X';
}
void dfs(int x, int y, vector<vector<char>>& board)
{
st[x][y] = true;
int dx[4] = {-1,0,1,0}, dy[4] = {0,1,0,-1};
for(int j = 0;j<4;j++)
{
int a = x+dx[j], b = y + dy[j];
if(a>=0 && a<n && b>=0 && b<m && !st[a][b] && board[a][b] == 'O')
dfs(a,b,board);
}
}
};