题目:
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
解释:拔下计算机 1 和 2 之间的线缆,并将它插到计算机 1 和 3 上。
示例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)
- connections[i].length == 2
- 0 <= connections[i][0], connections[i][1] < n
- connections[i][0] != connections[i][1]
- 没有重复的连接。
- 两台计算机不会通过多条线缆连接。
解题思路:
先初始化待连接的节点数为 n-1 ,遍历connections时,若网线两头的节点已经在一个集合中,那么这条网线是多余的,然后 lable 加一;如果不在一个集合中,就执行union函数。然后res减一。最后将res和lable进行比较,如果lable大于res就代表多余的网线足够用来连接待连接的节点,否则返回-1.
代码:
class Solution:
def makeConnected(self, n: int, connections: List[List[int]]) -> int:
res = n - 1
lable = 0
lst = [i for i in range(n)]
def find(x):
if lst[x] != x:
lst[x] = find(lst[x])
return lst[x]
def union(x, y):
lst[find(y)] = find(x)
for a in connections:
x, y = a[0], a[1]
if find(x) == find(y):
lable += 1
else:
union(x, y)
res -= 1
return res if res <= lable else -1
提交记录: