力扣刷题笔记：947. 移除最多的同行或同列石头（并查集，很容易理解的代码、完整题解代码及注释）

947、移除最多的同行或同列石头
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
不会有两块石头放在同一个坐标点上

解题思路:

看完题目，第一感觉就是找出多少组连通的石头。
而每组连通的石头需要保留至少一个石头，所以答案就是石头个数-连通石头的组数。

看完官方题解之后，明白了一个技巧，把所有坐标的纵坐标加上10000（坐标的最大值），我们可以把坐标看成edges。
举一个例子，[[0,0],[1,1],[1,2]] => [[0,10000],[1,10001],[1,10002]。
这里0, 10000, 1, 10001, 10002表示5个nodes。
[0,10000]表示node 0和node 10000有edge连通。
[1,10001]表示node 1和node 10001有edge连通。
[1,10002]表示node 1和node 10002有edge连通。
这就转成熟悉的图并查集问题了，遍历所有edges，即可找出所有连通组。

时间复杂度O(nlogm)，n为edges数量，m为node数量。
空间复杂度O(m),这里使用了字典作实现并查集的，当然列表也是OK的。

此外，注意在最后要对所有node调用find，更新所属的连通组。

并查集python模板：https://blog.csdn.net/weixin_44414948/article/details/114002727

题解代码：

class Solution:
    def removeStones(self, stones: List[List[int]]) -> int:
    	# 以字典形式定义初始化的parent集
        dus ={s+i*10000:s+i*10000 for stone in stones for i, s in enumerate(stone)}
        print(dus)
		# 定义查找祖值方法（含更新祖值作用）
        def find(i):
            if dus[i] != i:
                dus[i] = find(dus[i])
            return dus[i]

		# 遍历全部石头，按顺序更新祖值
        for i, (s1, s2) in enumerate(stones):
            if s1 in dus and find(s1) != find(s2+10000):
                # union
                dus[find(s1)] = find(s2+10000)
                print(i, dus)
        # 防止按顺序更新的祖集中有重复未更新，再次更新
        for k in dus:
            print(k)
            find(k)
        # 最后用总数减去不重复的祖值数即可
        return len(stones) - len(set(dus.values()))

输出结果：

输入：

[[0,1],[0,2],[4,3],[2,4],[0,3],[1,1]]

输出：

4

stdout：

{0: 0, 10001: 10001, 10002: 10002, 4: 4, 10003: 10003, 2: 2, 10004: 10004, 1: 1}
0 {0: 10001, 10001: 10001, 10002: 10002, 4: 4, 10003: 10003, 2: 2, 10004: 10004, 1: 1}
1 {0: 10001, 10001: 10002, 10002: 10002, 4: 4, 10003: 10003, 2: 2, 10004: 10004, 1: 1}
2 {0: 10001, 10001: 10002, 10002: 10002, 4: 10003, 10003: 10003, 2: 2, 10004: 10004, 1: 1}
3 {0: 10001, 10001: 10002, 10002: 10002, 4: 10003, 10003: 10003, 2: 10004, 10004: 10004, 1: 1}
4 {0: 10002, 10001: 10002, 10002: 10003, 4: 10003, 10003: 10003, 2: 10004, 10004: 10004, 1: 1}
5 {0: 10002, 10001: 10003, 10002: 10003, 4: 10003, 10003: 10003, 2: 10004, 10004: 10004, 1: 10003}
0
10001
10002
4
10003
2
10004
1

作者：jasoncoding13
链接：https://leetcode-cn.com/problems/most-stones-removed-with-same-row-or-column/solution/pythonbing-cha-ji-onlogm94om19-by-jasonc-3uv7/
来源：力扣（LeetCode）https://leetcode-cn.com/problems/most-stones-removed-with-same-row-or-column/