复旦大学2017复试上机第三题——无向图的最小权值连通子图（prim算法实现）

复旦大学2017复试上机第三题——无向图的最小权值连通子图（prim算法的邻接矩阵实现）

题目：
一个无向图，顶点为N个，其中M条边已给定，现在要从K条备选边中选出若干条，使得整个图连通，且选出的边权值和最小。
（本作者自定义实验数据测试代码正确性）
4 4
1 2 2
1 4 1
2 3 3
3 4 4
第一行的两个数据分别为N顶点个数和边M的个数；下面的M行为每条边的数据，
起始点和终点，还有每条边的权值。本数据使整个图连通的最小权值之和为6.

#include <cstdio>
#include <algorithm>

using namespace std;

const int INF =100000;
const int maxn =100;
// 节点编号从1开始，邻接矩阵实现prim算法


int G[maxn][maxn]; bool vis[maxn]; int d[maxn]; 
// G矩阵存储任意两点间的距离，d存储从起始点到任意各点的最短距离

int n,m; //n为顶点数目，m为边的数目

int prim(int v)  //普利姆算法求最小连通图
{
    int sum=0;//最短路径长度

    fill(vis,vis+maxn,false);
    fill(d,d+maxn,INF);

    d[v]=0;  //到达本身的距离为0

    for(int i=1;i<=n;i++)
    {
        int u =-1;int mins = INF;
        for(int j=1;j<=n;j++)
        {
            if(vis[j]==false && d[j]<mins)
            {
                u =j; mins= d[j];
            }
        }
        if(u==-1) return -1;  //代表图不连通，出错
        else
        {
            vis[u] = true;
            sum +=d[u];

            for(int j=1;j<=n;j++)
            {
                if(vis[j]==false && G[u][j]!=INF  && G[u][j] <d[j])
                    d[j] = G[u][j];
            }
        }
    }
    return sum;
}

int main()
{
    scanf("%d %d",&n,&m);

    fill(G[0],G[0]+maxn*maxn,INF);

    for(int i=1;i<=m;i++)
    {
        int u,v,len; scanf("%d %d %d",&u,&v,&len);

        G[u][v] = G[v][u] =len;
    }

    int sum =prim(1);

    printf("%d\n",sum);
    return 0;
}

prim算法的邻接矩阵实现代码如下