添加了684.冗余连接 的Python并查集简洁写法

This commit is contained in:
CH Ye
2023-09-15 16:09:38 -04:00
parent a0462c4c25
commit 23ca9e43e5

View File

@ -256,6 +256,23 @@ class Solution:
return []
```
### Python简洁写法
```python
class Solution:
def findRedundantConnection(self, edges: List[List[int]]) -> List[int]:
n = len(edges)
p = [i for i in range(n+1)]
def find(i):
if p[i] != i:
p[i] = find(p[i])
return p[i]
for u, v in edges:
if p[find(u)] == find(v):
return [u, v]
p[find(u)] = find(v)
```
### Go
```go