贪心算法基础之最小生成树prim算法 51nod prim模板

题目：n个地点。现在给出m组地点之间的距离。

现在在这n个地点之间铺设网线，使得网线用的最少（要求每个地点都能与其他点接通）。

输入

第1行：2个数N,M中间用空格分隔，N为点的数量，M为边的数量。（2 <= N <= 1000, 1 <= M <= 50000)第2 - M + 1行：每行3个数S E W，分别表示M条边的2个顶点及权值。(1 <= S, E <= N，1 <= W <= 10000)

输出

输出最小生成树的所有边的权值之和。

输入示例

9 141 2 42 3 83 4 74 5 95 6 106 7 27 8 18 9 72 8 113 9 27 9 63 6 44 6 141 8 8

输出示例

37

分析：

最小生成树，经典算法prim算法。

令集合A表示已经连接的点，用集合B表示未连接的点。

1、一开始，我们随机的从一个点开始走，不妨由点1开始。

           A = { 1 };    B={ 2 , 3 , 4 , 5 ......}；

2、然后，找一个距离1最近的点，连线。假如这个点是4，那就在1,4之间连线

         A = { 1 ，4 };    B={ 2 , 3 , 5 ......}；

3、然后，从集合A，B中各取一个点，使得这两点的距离是最短的。假设是4和2之间最短

         A = { 1 ，4 ，2 };    B={ 3 , 5 ......}；

重复3的过程，直到集合B为空。

基本思想就是这样，代码实现实际上就是把迪杰斯特拉算法的模板稍微做一些更改，但意义发生很大变化

dis【i】不再记录由起点到i的最短距离，而是距离i最近的前驱点，前驱点来自集合A

代码：

#include<stdio.h>#include<string.h>const int INF=5555555;int map[1010][1100];int dis[1100];//记录实时路长 int vis[1100];//标记访问 int n,m;void prim(int s){for(int i=1;i<=n;i++){dis[i]=map[s][i];vis[i]=0;}vis[s]=1;for(int i=2;i<=n;i++){int min=INF;int k=0;for(int j=1;j<=n;j++){if(!vis[j]&&min>dis[j])//找一条以j结尾的未连线的最短线段{min=dis[j];k=j;}}vis[k]=1;for(int j=1;j<=n;j++){if(!vis[j]&&dis[j]>map[k][j])//k到j的距离要更新{dis[j]=map[k][j];}}}}int main(){while(~scanf("%d%d",&n,&m)){for(int i=0;i<1001;i++)for(int j=0;j<1001;j++)if(i==j)map[i][j]=0;elsemap[i][j]=INF;while(m--){int s,t,w;scanf("%d%d%d",&s,&t,&w);map[s][t]=map[t][s]=w;}int sum=0;prim(1);for(int i=1;i<=n;i++)sum+=dis[i];printf("%d\n",sum);}return 0;}