题目
用以太网线缆将 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]
没有重复的连接。
两台计算机不会通过多条线缆连接。
解答
关于连通网路的,毫无疑问,可以使用并查集。这样的话,我们只要在初始化的时候,给并查集的类定义一个count,表示当前的连通量。以及一个多余的次数rest,即默认一开始有(n)个网络线缆,每次在调用Union函数的时候,如果不用进行合并操作,则rest加1,代表着可以剩余一个线缆。而按照题目,最后还必须将每个单独的连通量进行合并,则count个连通量需要count-1个线缆。因此,最后结果只要返回以上即可。
代码
class UF
{
private:
vector<int> uf;
vector<int> rank;
int count;
int rest;
public:
UF(int n):uf(vector<int>(n)),rank(vector<int>(n)),count(n),rest(0)
{
for(int i=0;i<n;++i)
uf[i]=i;
}
int Find(int index)
{
return uf[index]==index?uf[index]:uf[index]=Find(uf[index]);
}
void Union(int index1,int index2)
{
int find1 = Find(index1);
int find2 = Find(index2);
if(find1!=find2)
{
if(find2<find1)
swap(find1,find2);
uf[find1]=find2;
if(rank[find1]==rank[find2]) rank[find2]+=1;
--count;
}
else ++rest;
}
int getCount()
{
return count;
}
int getRest()
{
return rest;
}
};
class Solution {
public:
int makeConnected(int n, vector<vector<int>>& connections) {
UF uf(n);
for(auto connection:connections)
{
uf.Union(connection[0], connection[1]);
}
if(uf.getRest()<uf.getCount()-1) return -1;
return uf.getCount()-1;
}
};