Kruskal最小生成树算法

Kruskal最小生成树算法

适用情况

以边为主线贪心搜索，判断连通图时需要用到并查集

时间复杂度为 （n 为边的数量）

思想

①【边按升序排序】将所有的边从小到大排序，然后依次取边判断是否成环（相连）

②【不连通时加边】若不相连则连接，并记录边的个数；否则不相连

③【控制边的个数】最后使边的个数达到 n-1 个即可得到最小生成树

---

import java.util.Arrays;
import java.util.Comparator;

public class Solution{
	class Node{ //从 u 到 v 点的花费是 w
		int u,v,w;
	}
	
	static int n, m;
	static Node[] node = new Node[100005];
	static int[] f = new int[10005];
		
	private static int kruskal() {
		for(int i=0; i<=n; i++)//并查集初始化
			f[i] = i;
		
		Comparator<Node> cmp = new Comparator<Node>() {
			@Override
			public int compare(Node o1, Node o2) {
				if(o1.w > o2.w)
					return 1;
				else if(o1.w < o2.w)
					return -1;
				return 0;
			}
		};
		Arrays.sort(node, 0, m, cmp);//m 为边数
		
		int ans = 0;
		int num = 0;
		for(int i=0; i<m; i++) {
			int x = find(node[i].u);
			int y = find(node[i].v);
			if(x != y || node[i].w < 0) {//后一个条件是题意
				f[x] = y;
				ans += node[i].w;
				num ++;
			}
		}
		if(num < n-1)
			return Integer.MAX_VALUE;
		return ans;//返回最小生成树的花费
	}

	private static int find(int x) {
		return x == f[x] ? f[x] : (f[x] = find(f[x]));
	}
}
	

---

如有错误或不合理的地方，敬请指正~

加油！！