LeetCode 1319. 连通网络的操作次数（中等）

最新推荐文章于 2022-07-08 08:19:09 发布

VictorKa

最新推荐文章于 2022-07-08 08:19:09 发布

阅读量125

点赞数

分类专栏： LeetCode 文章标签： leetcode java

本文链接：https://blog.csdn.net/VictorKa/article/details/113044869

版权

LeetCode 专栏收录该内容

44 篇文章 1 订阅

订阅专栏

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

我的Java代码：

思路：这个问题实际上就是一个无向连通图的问题。要把 n 个结点连起来，至少需要 n-1 条边，如果边数小于 n-1 那就无法完成操作，返回 -1 。而对于边数足够的情况下，要用最少的操作，即移动最少的边，就要看图中有几个连通子图，如果只有m个连通子图，那么用 m-1 条边即可把这 m 个子图连成一整个连通图。这里考虑用并查集来存储图中结点和其连通关系，并得到图中连通子图的个数。

class Solution {
    public int makeConnected(int n, int[][] connections) {
		if(connections.length < n-1) {
			return -1;
		}
		int len = connections.length;
		UnionFind unionfind = new UnionFind(n); 
		for(int i = 0;i < len;i++) {
			int x = connections[i][0];
			int y = connections[i][1];
			if(!unionfind.isConnected(x, y)) {
				unionfind.union(x, y);
			}
		}
		int num = unionfind.getNum(n);
        return num-1;
    }
	private class UnionFind{
		private int[] fathers;
		private int[] ranks;

		public UnionFind(int n) {
			fathers = new int[n];
			ranks = new int[n];
			for(int i = 0;i < n;i++) {
				fathers[i] = i;
				ranks[i] = 1;
			}
		}
		public void union(int x, int y) {
			int rootX = find(x);
			int rootY = find(y);
			if(rootX == rootY) {
				return;
			}
			if(ranks[rootX] > ranks[rootY]) {
				fathers[rootY] = rootX;
			}else {
				fathers[rootX] = rootY;
			}
            if(ranks[rootX] == ranks[rootY]){
                ranks[rootY]++;
            }
		}
		public int find(int x) {
			if(x != fathers[x]) {
				fathers[x] = find(fathers[x]);
			}
			return fathers[x];
		}
		public boolean isConnected(int x, int y) {
			int rootX = find(x);
			int rootY = find(y);
			return (rootX==rootY);
		}
		public int getNum(int n) {
			int num = 0;
			for(int i = 0;i < n;i++) {
				if(i == find(i)) {
					num++;
				}
			}
			return num;
		}
	}
}