最小生成树与Prim算法

Posted uninstalllingyi

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了最小生成树与Prim算法相关的知识,希望对你有一定的参考价值。

最小生成树(MST)

定义

首先是一棵树(废话

其次没有回路(废话

包含全部顶点和V-1条边

边的权重和最小!!!!!

所以如果是单棵最小生成树,至少说明图是连通的。不然就是森林。

生成思路

既然是根据图生成树,那么至少要有遍历图。那么,便要从一个源点出发,来一场愉快的深搜或广搜。

深搜生成就叫DFS树(深度优先搜索树

广搜生成就叫BFS树(广度优先搜索树

我们只需要在if语句中,在递归调用语句之前做一点手脚,便可以达到目的。

Prim算法——让一棵小树长大

(别装了!Dijkstra我知道是你!)

时间复杂度

O(n^2),是根据图生成最小树的算法。

算法思路

是一个穿了马甲的Dijkstra算法。用的是蓝白点的思想。

Dijkstra是啥?https://www.cnblogs.com/Uninstalllingyi/p/10417446.html

每次循环都把一个蓝点u变成白点。并且这个蓝点u与白点相连的边权势当前所有蓝点中最小的min[u]

仔细想想,是贪心的思路哟…

好的,我们来手工模拟一下。求下面这个图的最小生成树。(里面是伪代码哟)

技术图片

(emm…其实这个是绿白点…没事你就假装一下色盲。)

初始所有的点全都是蓝点,所以说这里有一个数组min[i]来表示一下。那么除了起点,其他的全部初始化为无穷大(0x3f)。即min[1]=0。权值之和用int MST=0来保存。

那么第一次循环就没什么说的了,min[1]=0是最小的蓝点,然后把1变白。然后枚举和a相连的蓝点,修改它们与白点相连的最小边权(就是min数组啦)

技术图片

min[2]=map[1][2]=2;
min[3]=map[1][3]=4;
min[4]=map[1][4]=7;

第二次循环找到min[2]是最小的蓝点,变白。然后更新相连的蓝点。bulabulabula。

技术图片

min[3]=map[2][3]=1; min[5]=map[2][5]=2;

第三次循环…找到min[3]=1是最小的那个…变白…更新蓝点最小值…

技术图片

 

min[4]=map[3][4]=1;

哦,这里要注意一下,因为min[5]=2<6,所以5不用更新。

 第四次循环找到点4,第五次循环找到点5.反正他们也没相连的蓝点了,所以直接变白叭。

技术图片

那么,最后得到的权值之和是…

min[1]+min[2]+min[3]+min[4]+min[5]=
0+2+1+1+2=6;
技术图片
以1为起点生成最小生成树,min[v]表示蓝点与白点相连的最小边权。
MST表示最小生成树的权值之和。
⑴初始化min[v]=∞(v≠1);min[1]=0;MST=0;
⑵for(int i=1;i<=n;i++){
        ①寻找min[u]最小的蓝点u
        ②将u标记为白点
        ③MST+=min[u]
        ④for(int v=1;v<=n;v++)//与白点u相连的所有蓝点v
            if(w[u][v]<min[v]) min[v]=w[u][v]//注:w[u][v]表示第u行第v个元素。第u行表示和白点u相连的所有的点。
}
⑶算法结束,MST即为最小生成树的权值之和。
高能伪代码

这里有一道洛谷的模板题

https://www.luogu.org/problemnew/show/P3366

#include<iostream>
#include<cstring>
#include<cstdio>
//prim算法 
using namespace std;
const int maxx=0x3f3f3f3f;
const int MAXN=5005;
int n,m,x,y,z,map[MAXN][MAXN],minn[MAXN],MST=0,vis[MAXN];
int main(){
     scanf("%d%d",&n,&m);
     memset(map,0x3f,sizeof(map));
     for(int i=1;i<=m;i++){
         scanf("%d%d%d",&x,&y,&z);
         if(map[x][y]>z){
             map[x][y]=map[y][x]=z;
         }
     }
     memset(minn,0x3f,sizeof(minn));
     minn[1]=0;
     for(int i=1;i<=n;i++){
         int u=0;
         for(int j=1;j<=n;j++){
             if(!vis[j]&&minn[u]>minn[j]){
                 u=j;
             }
         }
         vis[u]=true;
         MST+=minn[u];
         for(int v=1;v<=n;v++){
             if(!vis[v]&&map[u][v]<minn[v]){
                 minn[v]=map[u][v];
             }
         }
     }
     printf("%d
",MST);
      
}

(倒是头一次知道洛谷如果不加cstdio的头文件是不可以用printf的…尴尬。

以上是关于最小生成树与Prim算法的主要内容,如果未能解决你的问题,请参考以下文章

最小生成树详解 prim+ kruskal代码模板

最小生成树及Prim算法及Kruskal算法的代码实现

最小生成树算法:Kruskal算法 Prim算法

HDU1102 最小生成树prim算法

最小生成树-Prim算法详解(含全部代码)

关于最小生成树的Prim算法和Kruskal算法