Leetcode 1319 连通网络的操作次数(并查集)

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]
  • 没有重复的连接。
  • 两台计算机不会通过多条线缆连接。

思路:并查集求联通块,答案为联通块数目-1,如果连边小于n-1则输出-1

class Solution {
public:
    int f[100005],cnt[100005];
    int fa(int x)
    {
        return f[x]==x?x:f[x]=fa(f[x]);
    }
    int makeConnected(int n, vector<vector<int>>& connections) {
        if(connections.size()<n-1)return -1;
        int ans=0;
        for(int i=0;i<n;i++)f[i]=i;
        for(int i=0;i<connections.size();i++)
        {
           int x=connections[i][0],y=connections[i][1];
            f[fa(x)]=fa(y);
        }
        for(int i=0;i<n;i++)cnt[fa(i)]++;
        for(int i=0;i<n;i++)
        {
            if(cnt[i])ans++;
        }
        //for(int i=0;i<n;i++)cout<<f[i]<<endl;
        return ans-1;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43798170/article/details/105426640