LeetCode 947. 移除最多的同行或同列石头 | 【Python】

947. 移除最多的同行或同列石头

---

题目来源：力扣（LeetCode）https://leetcode-cn.com/problems/most-stones-removed-with-same-row-or-column/

题目

---

n 块石头放置在二维平面中的一些整数坐标点上。每个坐标点上最多只能有一块石头。

如果一块石头的 同行或者同列 上有其他石头存在，那么就可以移除这块石头。

给你一个长度为 n 的数组 stones ，其中 stones[i] = [xi, yi] 表示第 i 块石头的位置，返回 可以移除的石子 的最大数量。

示例 1：

输入：stones = [[0,0],[0,1],[1,0],[1,2],[2,1],[2,2]]
输出：5
解释：一种移除 5 块石头的方法如下所示：
1. 移除石头 [2,2] ，因为它和 [2,1] 同行。
2. 移除石头 [2,1] ，因为它和 [0,1] 同列。
3. 移除石头 [1,2] ，因为它和 [1,0] 同行。
4. 移除石头 [1,0] ，因为它和 [0,0] 同列。
5. 移除石头 [0,1] ，因为它和 [0,0] 同行。
石头 [0,0] 不能移除，因为它没有与另一块石头同行/列。

示例 2：

输入：stones = [[0,0],[0,2],[1,1],[2,0],[2,2]]
输出：3
解释：一种移除 3 块石头的方法如下所示：
1. 移除石头 [2,2] ，因为它和 [2,0] 同行。
2. 移除石头 [2,0] ，因为它和 [0,0] 同列。
3. 移除石头 [0,2] ，因为它和 [0,0] 同行。
石头 [0,0] 和 [1,1] 不能移除，因为它们没有与另一块石头同行/列。

示例 3：

输入：stones = [[0,0]]
输出：0
解释：[0,0] 是平面上唯一一块石头，所以不可以移除它。

提示：

• 1 <= stones.length <= 1000
• 0 <= xi, yi <= 104
• 不会有两块石头放在同一个坐标点上

解题思路

---

思路：并查集

先审题，题目中给定一个二维平面，其中二维平面上的一些整数坐标点会放置一个石头，同个坐标点最多能放置一块石头。

现在题目有这样的规则：

• 如果一块石头的 同行或者同列 上有其他石头存在，那么就可以移除这块石头。

题目中，放置石头的坐标点由数组 $stones$ 给出。其中 $stones[i]=[x_i,y_i]$ 中 $x_i,y_i$ 分别表示石头的横纵坐标。

题目要求返回结果：

• 返回 可以移除的石子 的最大数量。

现在先看示例 1，用图的形式来看下：

输入：stones = [[0,0],[0,1],[1,0],[1,2],[2,1],[2,2]]
输出：5
解释：一种移除 5 块石头的方法如下所示：
1. 移除石头 [2,2] ，因为它和 [2,1] 同行。
2. 移除石头 [2,1] ，因为它和 [0,1] 同列。
3. 移除石头 [1,2] ，因为它和 [1,0] 同行。
4. 移除石头 [1,0] ，因为它和 [0,0] 同列。
5. 移除石头 [0,1] ，因为它和 [0,0] 同行。
石头 [0,0] 不能移除，因为它没有与另一块石头同行/列。

图示：

根据上图，我们如果将石块当成图的顶点时，那么当石块的横坐标或者纵坐标相同时，那么可用一条边将其连接在一起。如下图。

现在根据示例 1 中解释部分来看，我们可以发现，在这个连通部分的石块中，最终会根据规则被移除得只剩下一块石头。移除的过程动图如下：

同样的，我们再看示例 2，

输入：stones = [[0,0],[0,2],[1,1],[2,0],[2,2]]
输出：3
解释：一种移除 3 块石头的方法如下所示：
1. 移除石头 [2,2] ，因为它和 [2,0] 同行。
2. 移除石头 [2,0] ，因为它和 [0,0] 同列。
3. 移除石头 [0,2] ，因为它和 [0,0] 同行。
石头 [0,0] 和 [1,1] 不能移除，因为它们没有与另一块石头同行/列。

这里我们同样用动图来看下移除的过程：

在这里，我们可以看到初始时，坐落在 $[1,1]$ 的石块，因为都没有与之相同的横纵坐标，所以自成一个连通分量，而坐落在其他四个坐标的石块，则因为横坐标或纵坐标相同，所以四个石块能够相互连通成另一个连通分量。

而根据移除的结果可以看到，最终只剩下两个石块。而示例 2 中，我们可以看到，这里例子的石块合并后有两个连通分量，而一个连通分量中的石块，根据规则都能够移除直到只剩一个石块。所以 最多可移除的石块数量 = 所有石块数量 - 连通分量的数目。

因为只关注连通性，这里我们考虑使用并查集。

在前面的分析中，只要石块的横坐标或纵坐标相同，便可以进行连通。也就是说，当石块落在 $[x,y]$，所有横坐标为 $x$ 和 所有纵坐标为 $y$ 的石块都应该合并在同一连通分量中。那么这里就会有个问题，如何区分横坐标和纵坐标？而且在并查集中，两者不能够相等。

在题目后面的提示中提及，$0\le x_i,y_i\le 10^4$ ，也就说只要令 $x$ 或 $y$ 加上 $10^4+1$ 映射到不重合的区间中，即可解决这个问题。

因为需要计算连通分量的个数，用 $count$ 进行记录，初始新建顶点时加 $1$，当进行合并时减 $1$。

具体的代码实现如下。

class UnionFind:
    def __init__(self):
        self.parent = {}
        self.count = 0
    
    def find(self, x):
        if x not in self.parent:
            self.parent[x] = x
            self.count += 1
        
        if x != self.parent[x]:
            self.parent[x] = self.find(self.parent[x])
        return self.parent[x]
    
    def union(self, x, y):
        x_root = self.find(x)
        y_root = self.find(y)
        if x_root == y_root:
            return
        self.parent[x_root] = y_root
        self.count -= 1
    
    def get_count(self):
        return self.count

class Solution:
    def removeStones(self, stones: List[List[int]]) -> int:
        uf = UnionFind()
        n = len(stones)
        for x, y in stones:
            uf.union(x, y+10001)
        return n-uf.get_count()

欢迎关注

---

公众号 【书所集录】

---

如有错误，烦请指出，欢迎指点交流。若觉得写得还不错，麻烦点个赞👍，谢谢。