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块石头的方法如下所示:
- 移除石头 [2,2] ,因为它和 [2,1] 同行。
- 移除石头 [2,1] ,因为它和 [0,1] 同列。
- 移除石头 [1,2] ,因为它和 [1,0] 同行。
- 移除石头 [1,0] ,因为它和 [0,0] 同列。
- 移除石头 [0,1] ,因为它和 [0,0] 同行。
石头 [0,0] 不能移除,因为它没有与另一块石头同行/列。
示例 2:
输入:stones = [[0,0],[0,2],[1,1],[2,0],[2,2]]
输出:3
解释:一种移除 3 块石头的方法如下所示:
- 移除石头 [2,2] ,因为它和 [2,0] 同行。
- 移除石头 [2,0] ,因为它和 [0,0] 同列。
- 移除石头 [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/