JAVA-prim 最小生成树

sd_WangYaJun

于 2017-08-03 16:43:24 发布

阅读量301

点赞数

CC 4.0 BY-SA版权

分类专栏： java-数据结构文章标签： prim 算法-图论

本文链接：https://blog.youkuaiyun.com/sd_WangYaJun/article/details/76636798

java-数据结构专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个使用Prim算法寻找最小生成树的Java程序实例。通过输入边的权重，程序能够计算并输出给定图的最小生成树的总权重。适用于理解与实践图论中最小生成树的概念。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package prim;

import java.util.Scanner;

public class Main {

    static int MAX=Integer.MAX_VALUE;
    public static void main(String[] args) {
        int n;
        int[][] A=new int[1001][1001];
        Scanner sin=new Scanner(System.in);
        while(sin.hasNext()){
            n=sin.nextInt();
            for(int i=0;i<n;i++){
                for(int j=0;j<n;j++){
                    A[i][j]=sin.nextInt();
                }
            }
            System.out.println(prim(A,n));
        }
    }

    public static int prim(int map[][],int n){
        int[] dis=new int[1001];
        int i, j, min, minID, sum=0;
        for(i=0;i<n;i++){
            dis[i]=map[0][i];
        }

        for(i=1;i<n;i++){
            min=MAX;
            minID=0;
            for(j=1;j<n;j++){
                if(dis[j]!=0 && dis[j]<min){
                    min=dis[j];
                    minID=j;
                }                   
            }
            if(min==0) return -1;
            sum+=min;
            dis[minID]=0;
            for(j=1;j<n;j++){
                if(dis[j]!=0 && dis[j]>map[minID][j]){
                    dis[j]=map[minID][j];
                }
            }

        }
        return sum; 
    }

}