一、注意点
该算法的原理同Dijkstra算法,也是不断将点加入集合的方式。
二、模板
//邻接表
struct Node{
int v,dis;//v是边的目标顶点,dis为边权
};
vector<Node> Adj[MAXV];//图G,Adj[u]存放从顶点u出发可以到达的所有点
int n;//n为顶点数,图G使用邻接表实现,MAXV为最大顶点数
int d[MAXV];//顶点与集合s的最短距离
bool vis[MAXV]={false};//标记数组,vis[i]==true表示已访问。初值false
int prim(){//默认0号为初始点,函数返回最小生成树的边权之和
fill(d,d+MAXV,INF);//整个数组赋初值INF
d[0]=0; //只有0号顶点到集合s的距离为0,其余全为INF
int ans=0; //存放最小生成树的边权之和
for(int i=0;i<n;i++)//循环n次
{
int u=-1,MIN=INF;//u用来存放节点,MIN用来比较距离
for(int j=0;j<n;j++)
//找到没有访问的顶点中,d[]最小的,即离集合最近的
{
if(vis[j]==false&&d[j]<MIN)
{
u=j; //找到距离最近的节点
MIN=d[j]; //更新最近的距离
}
}
//如果找不到小于INF(MIN)的d[u],则剩下的顶点和集合S不连通
if(u==-1)return -1;//返回失败
//若找到了
vis[u]=true;//标记u为已经访问
ans+=d[u];//将该点加入集合,并将边权累加
//对加入集合的点能够到达的其他点的d进行更新
for(int j=0;j<Adj[u].szie();j++)
{
int v=Adj[u][j].v;//通过邻接表得到u能够到达的顶点v
if(vis[v]==false&&Adj[u][j].dis<d[v])
{
//如果v没有访问&&以u为中介点可以使得v离集合更近
d[v]=Adj[u][j;
}
}
}
return ans;
}
三、适用问题
prim适用于稠密图,即图的顶点较少,边数较多的情况。