技術標籤：Python

題目

用乙太網線纜將 n 臺計算機連線成一個網路，計算機的編號從 0 到 n-1。線纜用 connections 表示，其中 connections[i] = [a, b] 連線了計算機 a 和 b。

網路中的任何一臺計算機都可以通過網路直接或者間接訪問同一個網路中其他任意一臺計算機。

給你這個計算機網路的初始佈線 connections，你可以拔開任意兩臺直連計算機之間的線纜，並用它連線一對未直連的計算機。請你計算並返回使所有計算機都連通所需的最少操作次數。如果不可能，則返回 -1 。

示例 1：

輸入：n = 4, connections = [[0,1],[0,2],[1,2]]

輸入：n = 6, connections = [[0,1],[0,2],[0,3],[1,2],[1,3]]
輸出：2
示例 3：

輸入：n = 6, connections = [[0,1],[0,2],[0,3],[1,2]]
輸出：-1
解釋：線纜數量不足。
示例 4：

輸入：n = 5, connections = [[0,1],[0,2],[3,4],[2,3]]
輸出：0

提示：

1 <= n <= 10^5
1 <= connections.length <= min(n*(n-1)/2, 10^5)

解題思路：

並查集，現學現用

class UnionFind:
    def __init__(self, n: int):
        self.parent = list(range(n))
        self.size = [1] * n
        self.n = n
        self.
 
setCount = n
    
    def findset(self, x: int) -> int:
        if self.parent[x] == x:
            return x
        self.parent[x] = self.findset(self.parent[x])
        return self.parent[x]
    
    def unite(self, x: int, y: int) -> bool:
        x, y = self.findset(x), self.findset(y)
        if x == y:
            return False
        if self.size[x] < self.size[y]:
            x, y = y, x
        self.parent[y] = x
        self.size[x] += self.size[y]
        self.setCount -= 1
        return True
    
    def connected(self, x: int, y: int) -> bool:
        x, y = self.findset(x), self.findset(y)
        return x == y



class Solution:
    def makeConnected(self, n: int, connections: List[List[int]]) -> int:
        if len(connections) < n-1: 
             return -1
        uf = UnionFind(n)
        for x,y in connections:
            uf.unite(x,y)
        return uf.setCount - 1