最小生成树 prim算法（附代码）

prim算法是以一个根节点开始慢慢往下延伸，不断寻找距生成树最短的距离的节点，然后将该节点纳入生成树的集合中，然后再将该节点影响的其他未纳入生成树节点的距离更新。（缩小与生成树的距离），重复操作，直至全部节点纳入集合或者没有节点纳入集合为止。

prim算法的时间复杂度为O（V^2）；

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int maxn=1005;
const int INF=0x3f3f3f3f;
int p[maxn]; //储存父节点
int edge[maxn][maxn];//邻接矩阵
int d[maxn]; //到生成树的最短距离
int vis[maxn];//是否加入集合中
int n,m;
//初始化
void init()
{
    memset (vis,0,sizeof(vis));
    memset (p,-1,sizeof(p));
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
             i==j? edge[i][j]==0:edge[i][j]=INF;
}
void prim()
{
    //以1作为根节点
    for (int i=1;i<=n;i++)
    {
        if(edge[1][i]!=INF)
            p[i]=1;
        d[i]=edge[1][i];
    }
    while (1)
    {
        int maxx=INF;
        int u=-1;
        //找到距离生成树最短距离的节点
        for (int i=1;i<=n;i++)
        {
            if(!vis[i]&&d[i]<maxx)
            {
                maxx=d[i];
                u=i;
            }
        }
        if(u==-1)
            break;
        //将选出的节点纳入集合
        vis[u]=1;
        //更新该节点影响的节点
        for (int i=1;i<=n;i++)
        {
            if(!vis[i]&&d[i]>edge[u][i])
            {
                d[i]=edge[u][i];
                p[i]=u;
            }
        }
    }
    int sum=0;
    for (int i=1;i<=n;i++)
        if(p[i]==-1)
        {
            printf("不存在最小生成树\n");
            return ;
        }
        else
            sum+=d[i];
        printf("最短生成树的长度为%d\n",sum);
        return ;
}
void parent ()
{
    for (int i=1;i<=n;i++)
        if(p[i]!=-1&&p[i]==i)
              printf("%d为根节点\n",i);
        else if(p[i]!=-1)
              printf("%d的父节点为%d\n",i,p[i]);
}
int main()
{
    scanf("%d%d",&n,&m);
    init();
    for (int i=1;i<=m;i++)
    {
        int x,y,sp;
        scanf("%d%d%d",&x,&y,&sp);
        edge[x][y]=edge[y][x]=sp;
    }
    prim();
    parent();
    return 0;
}
/*运行结果
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
最短生成树的长度为12
1为根节点
2的父节点为5
3的父节点为6
4的父节点为3
5的父节点为3
6的父节点为1*/