1319. 连通网络的操作次数(傻瓜教程)(python)(LC)

1319. 连通网络的操作次数

用以太网线缆将 n 台计算机连接成一个网络,计算机的编号从 0 到 n-1。线缆用 connections 表示,其中 connections[i] = [a, b] 连接了计算机 a 和 b。

网络中的任何一台计算机都可以通过网络直接或者间接访问同一个网络中其他任意一台计算机。

给你这个计算机网络的初始布线 connections,你可以拔开任意两台直连计算机之间的线缆,并用它连接一对未直连的计算机。请你计算并返回使所有计算机都连通所需的最少操作次数。如果不可能,则返回 -1 。


示例 1:

输入:n = 4, connections = [[0,1],[0,2],[1,2]]
输出:1
解释:拔下计算机 12 之间的线缆,并将它插到计算机 13 上。

示例 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

思路:并查集

一、因为涉及到0-(n-1)中每一个元素,所以我们无需添加add()方法,且可以采用并查集模版暴力构图

二、对“拔线”的理解体现为对是否构成“环”对判断,为了方便,我们取消isconnected()方法,并把它插入union()方法中

三、添加“more”元素作为结果的衡量,本质上,“more”是对union()方法中isconnected()元素的计数

四、根据题意输出结果

完整代码

class UF:
    def __init__(self, n):
        self.parent = {
    
    i: i for i in range(n)}
        self.cnt = n
        self.more = 0

    def find(self, x):
        root = x
        while self.parent[root] != root:
            root = self.parent[root]
        while x != root:
            original_parent = self.parent[x]
            self.parent[x] = root
            x = original_parent
        return root

    def union(self, x, y):
        leader_x = self.find(x)
        leader_y = self.find(y)
        if leader_x != leader_y:
            self.parent[leader_x] = leader_y
            self.cnt -= 1
        else:
            self.more += 1

class Solution:
    def makeConnected(self, n: int, connections) -> int:
        uf = UF(n)
        for x, y in connections:
            uf.union(x, y)
        if uf.more >= uf.cnt - 1:
            return uf.cnt -1
        elif uf.more < uf.cnt - 1:
            return -1

猜你喜欢

转载自blog.csdn.net/qq_51174011/article/details/113061576