使用并查集实现Kruscal算法

package ACM.Graph;

/* 使用并查集实现Kruscal算法   
 * */
import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

class Edge {
	int num;// 边的编号
	int u;// 边的起点
	int v;// 边的终点
	int w;// 边的权重;

	public Edge(int num, int u, int v, int w) {
		this.num = num;
		this.u = u;
		this.v = v;
		this.w = w;
	}
}

class EdgeComparator implements Comparator<Edge> {
	@Override
	public int compare(Edge o1, Edge o2) {
		if (o1.w < o2.w)
			return -1;
		else if (o1.w > o2.w)
			return 1;
		return 0;
	}
}

public class Kruscal {

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();// 顶点数
		int m = scanner.nextInt();// 边数
		Edge[] e = new Edge[m];
		for (int i = 0; i < m; i++) {
			e[i] = new Edge(i, scanner.nextInt(), scanner.nextInt(), 
					scanner.nextInt());
		}
		EdgeComparator ec = new EdgeComparator();
		Arrays.sort(e, ec);//按权值大小给边排序
		int[] p = new int[n + 1];
		for (int i = 1; i <= n; i++)// 初始化并查集
			p[i] = i;
		int ans = 0;
		Kruscal kru = new Kruscal();
		for (int i = 0; i < m; i++) {
			Edge ee = e[i];
			int x = kru.find(ee.u, p);// 找出当前边两个端点的编号
			int y = kru.find(ee.v, p);
			if (x != y) {// 不在同一个并查集就合并
				System.out.println("edge:(" + ee.u + "," + ee.v + ")");
				ans += ee.w;
				p[x] = y;
			}
		}
		System.out.println("The minmum price is " + ans);
	}

	private int find(int i, int[] p) {
		if (p[i] == i)
			return i;
		p[i] = find(p[i], p);// 更改i的父节点为根
		return p[i];
	}
}

输入
9
14
1 2 4
2 3 8
3 4 7
4 5 9
5 6 10
6 7 2
7 8 1
8 1 8
2 8 11
3 9 2
9 8 7
9 7 6
3 6 4
4 6 14
输出
edge:(7,8)
edge:(6,7)
edge:(3,9)
edge:(1,2)
edge:(3,6)
edge:(3,4)
edge:(2,3)
edge:(4,5)
The minmum price is 37


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值