最小生成树prim

public class prim {
	static int M = 1000;

	public static void main(String[] args) {
		int graph[][] = { { M, 6, 1, 5, M, M }, { 6, M, 5, M, 3, M },
				{ 1, 5, M, 5, 6, 4 }, { 5, M, 5, M, M, 2 },
				{ M, 3, 6, M, M, 6 }, { M, M, 4, 2, 6, M } };
		int start = 0;
		int n = graph.length;
		int[] visited = new int[n];
		System.out.println(prim(graph, visited, start, n));
	}

	public static int prim(int[][] graph, int[] visited, int start, int n) {
		int pos, result = 0;
		int min = Integer.MAX_VALUE;
		int[] low = new int[n];
		// 从某点开始,分别标记和记录该点
		visited[start] = 1;
		pos = start;
		System.out.print(pos + " ");
		// 第一次给low数组赋值
		for (int i = 0; i < n; i++)
			if (i != pos)
				low[i] = graph[pos][i];
		// 再运行n-1次
		for (int i = 0; i < n - 1; i++) {

			// 找出最小权值并记录位置
			min = Integer.MAX_VALUE;
			for (int j = 0; j < n; j++)
				if (visited[j] == 0 && min > low[j]) {
					min = low[j];
					pos = j;//记录最小值点
				}
			// 最短路径
			result += min;
			// 求出最小值后,设置为已经访问
			visited[pos] = 1;
			// 更新权值
			for (int j = 0; j < n; j++)
				if (visited[j] == 0 && low[j] > graph[pos][j])
					low[j] = graph[pos][j];
			System.out.print(pos + " ");
		}
		return result;
	}
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值