最小生成树, prim 算法
普里姆算法(Prim算法),图论中的一种算法,可在加权连通图里搜索最小生成树.
下面对算法的图例描述(转载 的图 , 用于理解) ; 其原理 类似于 Dijstra算法最短路算法
图例 | 说明 | 不可选 | 可选 | 已选(Vnew) |
---|---|---|---|---|
此为原始的加权连通图。每条边一侧的数字代表其权值。 | - | - | - | |
顶点D被任意选为起始点。顶点A、B、E和F通过单条边与D相连。A是距离D最近的顶点,因此将A及对应边AD以高亮表示。 | C, G | A, B, E, F | D | |
下一个顶点为距离D或A最近的顶点。B距D为9,距A为7,E为15,F为6。因此,F距D或A最近,因此将顶点F与相应边DF以高亮表示。 | C, G | B, E, F | A, D | |
算法继续重复上面的步骤。距离A为7的顶点B被高亮表示。 | C | B, E, G | A, D, F | |
在当前情况下,可以在C、E与G间进行选择。C距B为8,E距B为7,G距F为11。E最近,因此将顶点E与相应边BE高亮表示。 | 无 | C, E, G | A, D, F, B | |
这里,可供选择的顶点只有C和G。C距E为5,G距E为9,故选取C,并与边EC一同高亮表示。 | 无 | C, G | A, D, F, B, E | |
顶点G是唯一剩下的顶点,它距F为11,距E为9,E最近,故高亮表示G及相应边EG。 | 无 | G | A, D, F, B, E, C | |
现在,所有顶点均已被选取,图中绿色部分即为连通图的最小生成树。在此例中,最小生成树的权值之和为39。 | 无 | 无 | A, D, F, B, E, C, G |
题目地址
代码实现
#include<stdio.h>
#include<iostream>
using namespace std;
const int inf=1<<29;
const int N= 1000;
int map[N][N];//邻接矩阵
int dist[N];// 到每个点的距离
int pre[N];//结点
int n;
int sum;
void pirm(int v0)
{
int i,j;
for(i=1;i<=n;i++)
{
dist[i]=map[v0][i];
pre[i]=i;
}
pre[v0]=0;
for(i=0;i<n-1;i++)
{
int mid=inf;
int u=-1;
for(j=0;j<=n;j++)
{
if(dist[j]&&dist[j]<mid)
{
mid=dist[j];
u=j;
}
}
sum+=dist[u];//计算路径之和
dist[u]=0;
for(j=0;j<=n;j++)
{
if(map[u][j]<dist[j])
{
dist[j]=map[u][j];
pre[j]=u;
}
}
}
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=0;i<=n;i++)//初始化
for(int j=0;j<=n;j++)
map[i][j]=inf;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",map[i]+j);//直接输入 or
sum=0;
pirm(1);//以1为起点
printf("%d\n",sum);
}
return 0;
}