最小生成树之Prim算法

Prim属于一种贪心算法，用于计算最小生成树。

将点集分为去过的和没去过的，所有去过的点都可以当作是起点，找当前离起点最近的点加入去过的点集。

#include<bits/stdc++.h>
#define MAXN 10010
#define INF 0x7fffffff
using namespace std;

int dis[MAXN],vis[MAXN];
int m[MAXN][MAXN];
int ans;                  //记录最小生成树所有边权重和

dis【i】数组用来记录到点i的距离，vis【i】用来记录是否到过i点，m【i】【j】记录点i到点j的距离

void Prim()
{
    int i,j,k,minn;
    d[1]=0;
    ans=0;
    for(i=1;i<=n;i++)   //选择N个点
    {
        minn=INF;
        for(j=1;j<=n;j++)
        {
            if(vis[j]==0&&dis[j]<minn)  //找到当前最近点
            {
                minn=dis[j];
                k=j
            }
        }
        vis[k]=1;  //记录
        ans+=dis[k];
        for(j=1;j<=n;j++)
        {
            if(vis[j]==0&&dis[j]>m[k][j])  //更新没去过的点的距离
            {
                dis[j]=m[k][j];
            }
        }
    }
}

主要分为两个部分，先找到最近的点，再用最近的点更新dis数组

int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        dis[i]=INF;
        vis[i]=0;
    }
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    {
         cin>>m[i][j];   
    }
    Prim();
    cout<<ans;
}

注意：

1. 这里的dis指到所有已去过点的最短距离，而dijkstra算法里的dis记录的是到起点的最短距离

2. 每次循环主要包括两部分，找到目前没去过点中最近的那一个；用找到的点更新dis