最小支撑（生成）树的prim算法 java

最新推荐文章于 2022-02-12 17:05:55 发布

文甜心

最新推荐文章于 2022-02-12 17:05:55 发布

阅读量311

点赞数

分类专栏：常用算法

本文链接：https://blog.csdn.net/qq_34893429/article/details/73717986

版权

常用算法专栏收录该内容

13 篇文章 0 订阅

订阅专栏

package prim;
import java.util.*;

public class Prim {
        static int MAX = Integer.MAX_VALUE;

        public static void main(String[] args) {
            int[][] map = new int[][] {
                    { 0, 10, MAX, MAX, MAX, 11, MAX, MAX, MAX },
                    { 10, 0, 18, MAX, MAX, MAX, 16, MAX, 12 },
                    { MAX, MAX, 0, 22, MAX, MAX, MAX, MAX, 8 },
                    { MAX, MAX, 22, 0, 20, MAX, MAX, 16, 21 },
                    { MAX, MAX, MAX, 20, 0, 26, MAX, 7, MAX },
                    { 11, MAX, MAX, MAX, 26, 0, 17, MAX, MAX },
                    { MAX, 16, MAX, MAX, MAX, 17, 0, 19, MAX },
                    { MAX, MAX, MAX, 16, 7, MAX, 19, 0, MAX },
                    { MAX, 12, 8, 21, MAX, MAX, MAX, MAX, 0 } };
            prim(map, map.length);
        }
         public static void prim(int[][] graph, int n){

                char[] nodeName = new char[]{'A','B','C','D','E','F','G','H','I'};       
                int[] lowCost = new int[n];  //到新集合的最小权  
                int[] mid= new int[n];//存取前驱结点 
                List<Character> list=new ArrayList<Character>();//用来存储加入结点的顺序 
                int i, j, min, minid , sum = 0;
                //初始化辅助数组 选取A点为第一个节点 
                for(i=1;i<n;i++)
                {
                    lowCost[i]=graph[0][i];
                    mid[i]=0;
                }
                list.add(nodeName[0]);
                //一共需要加入n-1个点 
                for(i=1;i<n;i++)
                {
                     min=MAX;
                     minid=0;
                     //每次找到距离集合最近的点 
                     for(j=1;j<n;j++)
                     {
                         if(lowCost[j]!=0&&lowCost[j]<min)
                         {
                             min=lowCost[j];
                             minid=j;
                         }
                     }

                     if(minid==0) return;//孤立点返回
                     list.add(nodeName[minid]);//将最小权值的点加入路径中
                     lowCost[minid]=0;//已经经过的点权值归0 防止出现圈
                     sum+=min;
                     System.out.println(nodeName[mid[minid]] + "到" + nodeName[minid] + " 权值：" + min);
                     //加入该点后，更新其它点到集合的距离 
                     for(j=1;j<n;j++)
                     {
                         if(lowCost[j]!=0&&lowCost[j]>graph[minid][j])//在新加入的点的权集中  与原有的权值进行取小   保证不成圈
                         {
                             lowCost[j]=graph[minid][j];
                             mid[j]=minid;
                         }
                     }
                }
                System.out.println("sum:" + sum);

            }

}