糖逗 发表于 2020-5-31 14:49:27

C++刷leetcode(1319. 连通网络的操作次数)【并查集】

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

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

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

 

示例 1:



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



输入:n = 6, connections = [,,,,]
输出:2
示例 3:

输入:n = 6, connections = [,,,]
输出:-1
解释:线缆数量不足。
示例 4:

输入:n = 5, connections = [,,,]
输出:0
 

提示:

1 <= n <= 10^5
1 <= connections.length <= min(n*(n-1)/2, 10^5)
connections.length == 2
0 <= connections, connections < n
connections != connections
没有重复的连接。
两台计算机不会通过多条线缆连接。

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


class Solution {
public:
    int find_root(int x, vector<int>& father){
      if(father != x) return find_root(father, father);
      return x;
    }
    void unite(int x, int y, vector<int>& father){
      int temp1 = find_root(x, father);
      int temp2 = find_root(y, father);
      if(temp1 != temp2){
            father = temp2;
      }
    }
    int makeConnected(int n, vector<vector<int>>& connections) {
      if(connections.size() < n-1) return -1;
      vector<int> father(n, 0);
      //初始化father
      for(int i = 0; i < n; i++){
            father = i;
      }
      //根据connections构建并查集
      int len = connections.size();
      for(auto cha : connections){
            unite(cha, cha, father);
      }
      //确定一共有几个group
      set<int> temp;
      for(int i = 0; i < n; i++){
            temp.insert(find_root(i, father));
      }
      int group = temp.size();
      return group - 1;
    }
};
页: [1]
查看完整版本: C++刷leetcode(1319. 连通网络的操作次数)【并查集】