最小生成树之Kruskal算法代码模板

常用算法代码模板总结

目录

算法实现思路

练习例题：Prim算法求最小生成树

---

算法实现思路

①记录图中每一条边，按照边的权值来对每一条边进行排序。

②枚举每一条边，如果这条边的起点和终点不在同一个集合（也就是a所在的集合和b所在的集合不连通），那么将这条边加入路径当中。

练习例题：Kruskal算法求最小生成树

 java题解

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

public class Main {

	static int m, n;
	static int[][] way;//记录边的信息
	static int[] p = new int[100010];
	static int INF = Integer.MAX_VALUE / 100;

	public static int Find(int x) {//并查集
		if (p[x] != x)
			p[x] = Find(p[x]);
		return p[x];
	}

	public static int kruskal() {

		for (int i = 1; i <= n; i++)//初始化每一个点，都是以自己为根节点的集合
			p[i] = i;

		int result = 0, count = 0;
		for (int i = 0; i < m; i++) {
			int a = way[i][0];
			int b = way[i][1];
			int c = way[i][2];
			a = Find(a);//查询a的根节点
			b = Find(b);//查询b的根节点
			if (a != b) {//如果a和b不在一个集合里面
				p[a] = b;//将a的根节点的父节点指向b
				result += c;//加上这条边的距离
				count++;//合并的点数加1
			}
		}

		if (count < n - 1)// 如果连通的点少于了点的总个数,说明不能构成最小生成树
			return INF;
		return result;

	}

	public static void main(String[] args) {
		Scanner input = new Scanner(System.in);
		n = input.nextInt();
		m = input.nextInt();
		way = new int[m][3];
		for (int i = 0; i < m; i++) {
			way[i][0] = input.nextInt();
			way[i][1] = input.nextInt();
			way[i][2] = input.nextInt();
		}
		Arrays.sort(way, new Comparator<int[]>() {//重写比较器，以边的权值作为比较依据
			public int compare(int[] o1, int[] o2) {
				return o1[2] - o2[2];
			}
		});

		int t = kruskal();
		if (t == INF)
			System.out.println("impossible");
		else
			System.out.println(t);

	}
}