最小生成树之prim算法代码模板

常用算法代码模板总结

目录

算法实现思路

练习例题：Prim算法求最小生成树

---

算法实现思路

①初始化所有的点到最小生成树集合的距离为正无穷。

②n次迭代，每次迭代找到当前不在最小生成树集合当中的距离集合最近的点，将这个点加入集合，然后用这个点来更新其他的点到集合的距离。

练习例题：Prim算法求最小生成树

java题解

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    public static int size = 510;//点的个数，多开10个防止越界
    public static int INF = 0x3f3f3f3f;//可以认为是正无穷
    public static int n, m;
    public static int[] dist = new int[size];//dist[i]存放i这个点到最小生成树集合的距离
    public static boolean[] st = new boolean[size];//st[i]表示这个点是否已经加入了集合
    public static int[][] map = new int[size][size];//存放图的关系

    public static void Init() {/
        Arrays.fill(dist, INF);//初始化点到集合的距离为正无穷
        for (int i = 0; i <= n; i++) {//初始化各个点之间的距离为正无穷
            Arrays.fill(map[i], INF);
        }
    }

    public static int prim() {
        int result = 0;//最小生成树集合大小
        for (int i = 0; i < n; i++) {//遍历n次,每次加入一个点
            int temp = -1;
            for (int j = 1; j <= n; j++) {//找到还没有加入集合当中距离集合最近的点
                if (st[j] == false && (temp == -1 || dist[temp] > dist[j]))
                    temp = j;
            }
            if (i != 0 && dist[temp] == INF)//如果这个点不是第一个点并且与集合的距离为正无穷，直接返回，因为最小的点距离集合的距离都为正无穷的，其他的点也是一样的
                return INF;
            if (i != 0) {//如果这个点不是第一个点，将这个点到集合的距离加入最小生成树
                result += dist[temp];
            }
            st[temp] = true;//将这个点标记为加入集合
            for (int j = 1; j <= n; j++)//用上面找到的点来更新其他点到集合的距离
                dist[j] = Math.min(dist[j], map[temp][j]);
        }
        return result;//返回结果
    }


    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        n = input.nextInt();
        m = input.nextInt();
        Init();
        for (int i = 0; i < m; i++) {
            int a = input.nextInt();
            int b = input.nextInt();
            int c = input.nextInt();
            map[a][b] = Math.min(map[a][b], c);//有重边
            map[b][a] = map[a][b];//两边都打通
        }
        int t = prim();
        if (t == INF) {//如果返回的结果为正无穷，表示没有生成最小生成树
            System.out.println("impossible");
        } else
            System.out.println(t);
    }

}