prim

最小生成树prim算法:

graph[][] 待生成树的图,

used[] 使用过的节点,

distance[] 生成树到各节点的最短距离,

start 任意起始点。

1:将 start 放入 used ,从起始点 start 遍历周边节点,将可到达的节点的距离放入 distance 。

public static void init(int[][] graph, boolean[] used, int[] distance){
		int from = GraphUtils.FROM;
		for (int i = 0; i < graph.length; i++){
			used[i] = false;
			distance[i] = graph[from][i];
		}
		used[from] = true;
		distance[from] = 0;
	}

2:在 distance 中选择没有在 used 中并且距离最小的联通点 min ,添加到 used 中。

3:遍历 min 联通的节点around,如果 graph[min][around] < distance[around] 设置 distance[around] = graph[min][around]。

注:此处可添加 sum+= min;用于计算最小生成树的路长总和。

4:循环 2,3 直到全部在 used 中。

public static int prim(int[][] graph){
		boolean[] used = new boolean[graph.length];
		int[] distance = new int[graph.length];
		init(graph, used, distance);
		int sum = 0;
		for (int i = 0; i < graph.length; i++){
			int min = GraphUtils.MAX;
			int minIndex = GraphUtils.FROM;
			for (int j = 0; j < graph.length; j++){
				if ((!used[j]) && distance[j] < min){
					min = distance[j];
					minIndex = j;
				}
			}
			if (min != GraphUtils.MAX) {
				used[minIndex] = true;
				sum += min;
				LOG.d(minIndex);
				for (int j = 0; j < graph.length; j++){
					if ((!used[j]) && distance[j] > graph[minIndex][j]){
						distance[j] = graph[minIndex][j];
					}
				}
			}
		}
		return sum;
	}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值