Prim算法求最小生成树
——HM
我来肤浅介绍一下Prim算法求最小生成树的方法。若有不知道最小生成树的OIer们可以点这里,附赠送Kruskal算法求最小生成树的方法。
Prim的中心思想主要为贪心的思想,大约过程如下:
1、任意选一点A;
2、将与A直接相连的点找出并比较其权值,取权值最小的一点B将其选入生成树的一点;
3、找出与线段AB直接相连且权值最小的点C并将其加入生成树;
4、重复上述步骤直至选入n个点。
不要在意我的语言表达能力,看不懂?用图来理解:
有一个如图所示的图:
任意选一点A将其入队,发现B、C、D与A直接相连
C离A最近将其入队,AC成为树的一条边,发现有B、D离与AC相连
CD<AB,D离AC最近,将D入队,CD成为树的一条边,有B、E与其相连。
DB<DE,B离D或A最近,将B入队,BD成为树的一条边,只剩E
EB<ED,选择BE,至此5个点全部入队,生成树节点依次为ACDBE,结束程序树如下图所示。
附标程(略丑勿喷):
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int SIZE=5001,INF=0x3f3f3f3f;
int n,e1,e;
int w[SIZE][SIZE];
int minc[SIZE];
void solve(int s)
{
int count=0,minv,k;
for(int i=1;i<=n;i++)minc[i]=w[s][i];
minc[s]=0;
for(int i=1;i<n;i++){
minv=INF;
for(int j=1;j<=n;j++){
if(minc[j] && minc[j]<minv){
minv=minc[j];
k=j;
}
}
minc[k]=0;
count+=minv;
for(int j=1;j<=n;j++){
if(w[k][j]<minc[j])
minc[j]=w[k][j];
}
}
printf("%d\n",count);
}
int main()
{
int t1,t2,t3;
for(int i=0;i<=SIZE;i++)
for(int j=0;j<=SIZE;j++)w[i][j]=INF;
scanf("%d%d",&n,&e);
for(int i=1;i<=e;i++){
scanf("%d%d",&t1,&t2,&t3);
if(t3<w[t1][t2])w[t2][t1]=w[t1][t2]=t3;
}
solve(n>>1);
return 0;
}