2020/11/09最小生成树 prim算法

最新推荐文章于 2023-12-07 23:17:01 发布

lsw518598

最新推荐文章于 2023-12-07 23:17:01 发布

阅读量353

点赞数

分类专栏： acwing学习

本文链接：https://blog.csdn.net/lsw518598/article/details/109575299

版权

acwing学习专栏收录该内容

37 篇文章 0 订阅

订阅专栏

题目

给定一个n个点m条边的无向图，图中可能存在重边和自环，边权可能为负数。

求最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

给定一张边带权的无向图G=(V, E)，其中V表示图中点的集合，E表示图中边的集合，n=|V|，m=|E|。

由V中的全部n个顶点和E中n-1条边构成的无向连通子图被称为G的一棵生成树，其中边的权值之和最小的生成树被称为无向图G的最小生成树。

输入格式
第一行包含两个整数n和m。

接下来m行，每行包含三个整数u，v，w，表示点u和点v之间存在一条权值为w的边。

输出格式
共一行，若存在最小生成树，则输出一个整数，表示最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

数据范围
1≤n≤500,
1≤m≤105,
图中涉及边的边权的绝对值均不超过10000。

输入样例：
4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4
输出样例：
6

源代码

import java.util.*;
class Main{
    static int N=510;
    static int INF=0x3f3f3f3f;
    static int n;
    static int m;
    static int[][] g=new int[N][N];
    static int[] dis=new int[N];
    static boolean[] st=new boolean[N];
    public static void main(String[] args){
        Scanner sc=new Scanner(System.in);
        n=sc.nextInt();
        m=sc.nextInt();
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i==j) g[i][j]=0;
                else g[i][j]=INF;
            }
        }
        while(m-->0){
            int x=sc.nextInt();
            int y=sc.nextInt();
            int z=sc.nextInt();
            g[x][y]=g[y][x]=Math.min(g[x][y],z);
        }
        int res=prim();
        if(res==-1) System.out.println("impossible");
        else System.out.print(res);
        
    }
    
    public static int prim(){
        Arrays.fill(dis,INF);
        int res=0;
        for(int i=0;i<n;i++){
            int t=-1;
            for(int j=1;j<=n;j++){
                if((!st[j])&&(t==-1||dis[j]<dis[t])) t=j;
            }
            if(i>0&&dis[t]==INF) return -1;
            if(i>0) res+=dis[t];
            st[t]=true;
            for(int j=1;j<=n;j++){
                dis[j]=Math.min(dis[j],g[j][t]);
            }
        }
        return res;
    }
}

算法注释

prim算法流程跟迪杰斯特拉算法类似,都是遍历+集合
变量定义: 遍历使用邻接矩阵g[][]
保存距离dis[]
确定节点是否在集合内st[]

核心思路:
有n个节点,共寻找n次(每次找一个){
找到距离集合最近的结点
标记/更新总长度
更新其他节点的距离(全更新即可,优秀的自然优秀不会受影响)
}
需要注意的是第一次寻找,第一次寻找时找到的那个点的距离一定是INF(因为所有点都是INF),所以这次的距离不能加入总路径长度;
同时,在寻找时我们也可以确定是否有最短生成树:
除了第一次寻找,剩下的寻找如果我们发现最小的距离是INF(说明剩下节点全是INF),那就表示后面的点都不能连接到这个集合,也就说明可return -1退出了.