目录

0130:被围绕的区域(★)

力扣第 130 题

题目

给你一个 m x n 的矩阵 board ,由若干字符 ‘X’‘O’ ,找到所有被 ‘X’ 围绕的区域,并将这些区域里所有的 ‘O’‘X’ 填充。

示例 1:

输入:board = [["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'。如果两个元素在水平或垂直方向相邻,则称它们是“相连”的。

示例 2:

输入:board = [["X"]]
输出:[["X"]]

提示:

  • m == board.length
  • n == board[i].length
  • 1 <= m, n <= 200
  • board[i][j]'X''O'

分析

  • 只有与边界连通的 ‘O’ 才不会被填充
  • 可以用 dfs 或 bfs 遍历找,更一般的是用并查集
    • 将边界上的 ‘O’ 都与哑节点 m*n 连通
    • 再将相邻的 ‘O’ 连通
    • 最后将不与哑节点连通的 ‘O’ 换成 ‘X’ 即可

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
class Solution:
    def solve(self, board: List[List[str]]) -> None:
        """
        Do not return anything, modify board in-place instead.
        """
        def find(x):
            if f[x]!=x:
                f[x]=find(f[x])
            return f[x]
        
        def union(x,y):
            f[find(x)]=find(y)

        m,n = len(board),len(board[0])
        f = list(range(m*n+1))
        A = [(i,j) for i in range(m) for j in range(n) if board[i][j]=='O']
        for i,j in A:
            if i in [0,m-1] or j in [0,n-1]:
                union(i*n+j,m*n)
            if i and board[i-1][j]=='O':
                union(i*n+j,(i-1)*n+j)
            if j and board[i][j-1]=='O':
                union(i*n+j,i*n+j-1)
        for i,j in A:
            if find(i*n+j)!=find(m*n):
                board[i][j] = 'X'

73 ms