算法设计： 四、贪心算法（2. 最小生成树 Prim 算法）—— java实现 - 算法分析

Prim 算法计算最小生成树

最小生成树的定义：最小生成树是一副连通加权无向图中一棵权值最小的生成树。假设给定无向图 G 一共有 n 个顶点，那么最小生成树一定会有 n-1 条边。

求解思想：

1. 选定一个点做为一个集合 a，剩下的点为另一个集合 b；
2. 将横跨两个集合且权重在其中最小的边加入最小生成树；
3. 将刚刚加入最小生成树的边中不在集合 a 中的点加入集合 a，直到所有的点加入集合 a。

流程图：

源码：

public class Prim {
	static int sum=0;
	static int M=10000;
	public static void main(String[] args) {
		int [][] c = {
			{0,0,0,0,0,0,0},
			{0,M,6,1,5,M,M},
			{0,6,M,5,M,3,M},
			{0,1,5,M,5,6,4},
			{0,5,M,5,M,M,2},
			{0,M,3,6,M,M,6},
			{0,M,M,4,2,6,M}
		};
		int n=c.length-1;
		System.out.println("依次构成树的边为（用两个顶点表示边）：");
		prim(n,c);
		System.out.println("最小权合值："+sum);
	}
	public static void prim(int n,int[][] c){
		float[] lowcost=new float[n+1];
		int[] closest=new int[n+1];//表示i到其他所有未添加进来的顶点的最短距离
		boolean[] s=new boolean[n+1];
     //初始化		
		s[1]=true;
		for(int i=2;i<=n;i++){
			lowcost[i]=c[1][i];
			closest[i]=1;
			s[i]=false;
		}
		for(int i=1;i<n;i++){//循环n-1次
			float min=Float.MAX_VALUE;
			int j=1;
			for(int k=2;k<=n;k++){
				if(lowcost[k]!=M&&(lowcost[k]<min)&&(!s[k])){
					min=lowcost[k];
					j=k;
				}
			}
			System.out.println(closest[j]+", "+j);
			s[j]=true;//将j添加到S中
			sum=sum+c[closest[j]][j];
			//逐个更改lowcost[k],如果c[j][k]<lowcost[k]则更改lowcost[k]为最小
			for(int k=2;k<=n;k++){
				if(!s[k]&&c[j][k]!=M){
					if(c[j][k]<lowcost[k]||lowcost[k]==M){
						lowcost[k]=c[j][k];
						closest[k]=j;
					}
				}
			}
		}
	}
}

算法分析

时间复杂度：O(n²)
空间复杂度：O(n)