[路飞]_前端算法第八十九弹-连通网络的操作次数

104 阅读2分钟

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

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

给你这个计算机网络的初始布线 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

这道题是需要将电脑连接起来,所以这道题依旧是需要用到并查集问题,而题目给到的是N台电脑和有限个网线,我们知道如果需要将N台电脑连接起来,那么至少需要N-1根网线,所以,如果现有网线数少于N-1根那么一定无法将其全部链接。现在我们需要将已经链接起来的电脑都保存在一个集合中,然后查看所有电脑一共存在于几个集合中,然后将其都存在一个集合中,将M个集合变成一个集合,需要移动至少M-1根。

所以我们就需要建立集合,整理集合。

同样需要先进行并查集建立

class UnionSet {
    constructor(n) {
        this.fa = new Array(n + 1)
        this.n = n
        for (let i = 0; i < n; i++) {
            this.fa[i] = i
        }
    }
    get(x) {
        return this.fa[x] = (this.fa[x] == x ? x : this.get(this.fa[x]))
    }
    merge(a, b) {
        this.fa[this.get(a)] = this.get(b)
    }
}

实现代码

var makeConnected = function (n, connections) {
    let len = connections.length
    if (n > len + 1) return -1
    let u = new UnionSet(n)
    for (let i = 0; i < len; i++) {
        const c = connections[i]
        u.merge(c[0], c[1])
    }
    let cnt = 0
    for (let i = 0; i < n; i++) {
        if (u.get(i) == i) cnt++
    }
    return cnt - 1
};