Java实现Prim算法,结合前向向量优化

Prim思想:
选择任意一个起点,这个点就是贪心策略刚开始的一个集合,这个集合就是一个整张图最小生成树的一部分(子集),然后通过贪心策略不断扩大这个子集。直到这个最小生成树集合包含了所有的点。
首先需要一个表,保存所有点到最开始最小生成树点集合的距离,不能到达就是INF;
然后,选择一个距离集合最近的一个点,加入集合(最小生成树点的集合),加入集合的点距离集合距离为0,下次不再计算;
通过这个新加入的点,更新集合到其它未加入的点的距离。
最后经过N-1步,就得到了最小生成树。

package com.yxs.graph1;

import java.io.BufferedInputStream;
import java.util.Arrays;
import java.util.Scanner;
import java.util.Stack;

public class Prim {
	static class Edge{
		int to,w,next;
		{
			to = 0;
			w = 0;
			next = 0;
		}
	}
	static final int MAXN = 105;
	static int[] head = new int[MAXN];
	static Edge[] edge = new Edge[MAXN];
	static int len = 1;
	
	static void add(int from, int to, int w) { 
		edge[len] = new Edge();
		edge[len].to = to;
		edge[len].w = w;
		edge[len].next = head[from];
		head[from] = len++;
	}
	
	static int[] dist = new int[MAXN];//dist[i]保存i到最小生成树集合最近的距离
	static int[] path = new int[MAXN];//path[i]保存顶点i接入最小生成树中的结点位置
	static final int INF = 0x3f3f3f3f;
	static {
		Arrays.fill(dist, INF);
	}
	
	static void prim(int start, int n) {
		for(int i = head[start];i != 0;i = edge[i].next) {
			dist[edge[i].next] = edge[i].w;
		}
		Arrays.fill(path, 1, n+1, start);
		dist[start] = 0;
		int ind, mind, sum = 0;
		//下面所说的集合内,表示已经生成好的最小生成树集合,最开始是一个点。
		//集合外所代表的是,还没有加入最小生成树集合的点
		for(int i = 1;i < n;i++) {//最小生成树只需要执行n-1步
			ind = 0;
			mind = INF;
			for(int j = 1;j <= n;j++) {//找一个当前最小生成树集合(最开始是一个点)能够到达的最近的点
				if(dist[j] != 0 && dist[j] < mind) {
					mind = dist[j];
					ind = j;
				}
			}
			if(mind == INF) break;//没有找到
			sum += mind;//计入最小生成树权值之和
			dist[ind] = 0;//加入最小生成树集合,集合扩大,下次不再使用该点
			for(int j = head[ind];j != 0;j = edge[j].next) {
				//通过找到的这个点,对其它在集合外的点进行松弛,看能不能通过找到的这个点缩短到集合外其它点的距离
				if(dist[edge[j].to] != 0 && edge[j].w < dist[edge[j].to]) {
					dist[edge[j].to] = edge[j].w;
					path[edge[j].to] = ind;
				}
			}	
		}
		System.out.println("最小生成树权值为:" + sum);
	}
	
	public static void main(String[] args) {
		Scanner in = new Scanner(new BufferedInputStream(System.in));
		int n = in.nextInt();
		int e = in.nextInt();
		for(int i = 1;i <= e;i++) {
			int from = in.nextInt();
			int to = in.nextInt();
			int w = in.nextInt();
			add(from, to, w);
		}
		prim(1, n);
	}

}
/*
6 8
1 6 100
1 5 30
1 3 10
2 3 5
3 4 50
4 6 10
5 4 20
5 6 60
*/


在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值