普里姆算法(Prim算法),图论中的一种算法,可在加权连通图里搜索最小生成树。意即由此算法搜索到的边子集所构成的树中,不但包括了连通图里的所有顶点(英语:Vertex (graph theory)),且其所有边的权值之和亦为最小。该算法于1930年由捷克数学家沃伊捷赫·亚尔尼克(英语:Vojtěch Jarník)发现;并在1957年由美国计算机科学家罗伯特·普里姆(英语:Robert C. Prim)独立发现;1959年,艾兹格·迪科斯彻再次发现了该算法。因此,在某些场合,普里姆算法又被称为DJP算法、亚尔尼克算法或普里姆-亚尔尼克算法。
(概念引自百度词条)
思路:不断更新点到最小生成树的距离,每次选一个到最小生成树距离最短的点加入最小生成树,并用加入最小生成树的点的边更定距离。直到所有的点都加入最小生成树,算法结束。
优化:存边的时候使用邻接表,时间复杂度由O(V^2)降为O(E*logV)。
模板:
/*
2016.8.6 BulaBulaCHN
*/
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<string>
#include<cstring>
#include<iostream>
using namespace std;
struct Edge
{
int from,to,val;
int next;
}eage[100005];
int tot=0;
int head[10005];
void add(int x,int y,int z)
{
eage[++tot].from=x;
eage[tot].to=y;
eage[tot].val=z;
eage[tot].next=head[x];
head[x]=tot;
}
int dis[10005];
int que[50005];
bool book[10005];
int tail=0;
int top=0;
int n,m;
int ans=0;
int main()
{
freopen("textdata.in","r",stdin);
freopen("textdata.out","w",stdout);
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
add(y,x,z);
}
que[++tail]=1;
book[1]=1;
for(int i=1;i<=n;i++) dis[i]=9999999;
dis[1]=0;
while(top<tail)
{
top++;
int po=9999999;
int point=-1;
for(int i=head[que[top]];i;i=eage[i].next)
if(!book[eage[i].to] && dis[eage[i].to]>eage[i].val)
dis[eage[i].to]=eage[i].val;
for(int i=1;i<=n;i++)
{
if(!book[i] && dis[i]<po)
{
po=dis[i];
point=i;
}
}
if(point!=-1)
{
book[point]=1;
que[++tail]=point;
ans+=po;
}
}
printf("%d",ans);
fclose(stdin);
fclose(stdout);
return 0;
}