力扣 leetcode 130. 被围绕的区域（python）

Topic

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

Example_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’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

Example_2

输入：board = [[“X”]]
输出：[[“X”]]

Tips

m == board.length
n == board[i].length
1 <= m, n <= 200
board[i][j] 为 ‘X’ 或 ‘O’

Solution_1

可以利用并查集解决本题
并查集模板

本题可以从边缘入手并将其想象为一个联通问题
首先先判断边缘的O
并找到所有与该O直接联通的O连接在并查集当中

之后需要遍历内部的元素
若遍历的元素其与已经连通在并查集中的O相连接
则将其也连通到并查集中
至此将所有连通的O连通在了并查集当中

最后再次遍历数组
把未连通的O均变为X

Code_1

class UnionFind:
    def __init__(self, n):
        self.father = list(range(n))
        self.size = [1] * n
    
    def find(self, x):
        if self.father[x] == x:
            return x
        self.father[x] = self.find(self.father[x])		
        return self.father[x]

    
    def merge(self, x, y):
        x, y = self.find(x), self.find(y)
        if x == y:
            return False
        if self.size[x]