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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-operations-to-make-network-connected
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

并查集

题解

class Solution {
void init(vector<int>& fa)
{
    for(int i=0;i<fa.size();i++){
        fa[i]=i;
    }
}
int find(vector<int>& fa,int x)
{
    if(fa[x]!=x){
        fa[x]=find(fa,fa[x]);
    }
    return fa[x];
}
public:
    int makeConnected(int n, vector<vector<int>>& connections) {
        if(connections.size()==0) return -1;
        vector<int> fa(n);
        init(fa);
        int cnt1=0,cnt2=0,cnt3=0,a=0,b=0;
        for(int i=0;i<connections.size();i++){
            a=find(fa,connections[i][0]);
            b=find(fa,connections[i][1]);
            if(a==b){
                cnt1++;
            }else{
                fa[a]=b;
                cnt2++;
            }
        }
        for(int i=0;i<n;i++){
            if(fa[i]==i){
                cnt3++;
            }
        }
        if(cnt1+1>=cnt3) return cnt3-1;
        else return -1;
    }
};