问题 H: 算法7-9：最小生成树 prim算法

 

题目描述

最小生成树问题是实际生产生活中十分重要的一类问题。假设需要在n个城市之间建立通信联络网，则连通n个城市只需要n-1条线路。这时，自然需要考虑这样一个问题，即如何在最节省经费的前提下建立这个通信网。

可以用连通网来表示n个城市以及n个城市之间可能设置的通信线路，其中网的顶点表示城市，边表示两个城市之间的线路，赋于边的权值表示相应的代价。对于n个顶点的连通网可以建立许多不同的生成树，每一棵生成树都可以是一个通信网。现在，需要选择一棵生成树，使总的耗费最小。这个问题就是构造连通网的最小代价生成树，简称最小生成树。一棵生成树的代价就是树上各边的代价之和。

而在常用的最小生成树构造算法中，普里姆（Prim）算法是一种非常常用的算法。以下是其算法的大致结构：

在本题中，读入一个无向图的邻接矩阵（即数组表示），建立无向图并按照以上描述中的算法建立最小生成树，并输出最小生成树的代价。

 

输入

输入的第一行包含一个正整数n，表示图中共有n个顶点。其中n不超过50。

以后的n行中每行有n个用空格隔开的整数，对于第i行的第j个整数，如果不为0，则表示第i个顶点和第j个顶点有直接连接且代价为相应的值，0表示没有直接连接。当i和j相等的时候，保证对应的整数为0。

输入保证邻接矩阵为对称矩阵，即输入的图一定是无向图，且保证图中只有一个连通分量。

输出

只有一个整数，即最小生成树的总代价。请注意行尾输出换行。

样例输入

4
0 2 4 0
2 0 3 5
4 3 0 1
0 5 1 0

样例输出

6

#include <stdio.h>
const int inf = 99999999;
int main()
{
    int n, data, sum, min, count;
    int map[55][55], vis[55], dis[55];
    while (~scanf("%d", &n))
    {
        sum = 0;
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {
                scanf("%d", &map[i][j]);
                if(i != j && !map[i][j])
                    map[i][j] = inf;
            }
        }
        int i=0;
        while(i<n)
        {
        	dis[i] = map[0][i];
            vis[i] = 0;
            i++;
		}
        vis[0] = count = 1;
        while (count < n)
        {
            min = inf;
            for (i = 0; i < n; i++)
            {
                if (!vis[i] && dis[i] < min)
                {
                    min = dis[i];
                    data = i;
                }
            }
            vis[data] = 1;
            count++;
            sum += dis[data];
            int k=0;
            while(k<n)
            {
            	if (!vis[k] && dis[k] > map[data][k])
                    dis[k] = map[data][k];
                    k++;
			}
        }
        printf("%d\n", sum);
    }
    return 0;
}


#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 1000+10
#define inf 0x3f3f3f3f3f
int map[maxn][maxn];
int vis[maxn];
int dis[maxn];
int main()
{
    int N,M;
    scanf("%d %d",&N,&M);
    memset(map,inf,sizeof(map));
    memset(vis,0,sizeof(vis));
    for(int i=0;i<M;i++)
    {
        int s,e,w;
        scanf("%d %d %d",&s,&e,&w);
        map[s][e]=min(w,map[s][e]);//多重边
        map[e][s]=min(w,map[e][s]);
    }
    for(int i=1;i<=N;i++)
    {
        dis[i]=map[1][i];
    }
    for(int i=1;i<=N;i++)
    {
        map[i][i]=0;
    }
    long long sum=0;
    vis[1]=1;
    int n=N-1;//共有N个点的无向图的最小生成树的边长为N-1
    int tip;
    while(n)
    {
        tip=0;//一个接一个的往下找
        for(int i=1;i<=N;i++)//寻找下一个最短的边//上一次记录的点周围的边中最短的边
        {
            if(!vis[i])
            {
                if(tip==0)
                    tip=i;
                else if(dis[i]<dis[tip])
                    tip=i;
            }
        }
        sum+=dis[tip];
        vis[tip]=1;
        n--;
        for(int i=1;i<=N;i++)
        {
            dis[i]=min(dis[i],map[tip][i]);//记录点tip周边的边长
        }
    }
    printf("%lld\n",sum);
    return 0;
}

ac代码

#include<iostream>
#include<cstring>
#define INF 99999999
using namespace std;
int map[55][55],dis[55];
bool vis[55];
int n;
void Prim()
{
	memset(dis,-1,sizeof(dis));
	memset(vis,0,sizeof(vis));
	dis[1]=0;
	int w,p,sum=0;
	for(int i=0;i<n;i++)
	{
		w=INF;
		for(int j=0;j<n;j++)
		{
			if(!vis[j]&&dis[j]<w&&dis[j]!=-1)
			{
				p=j;
				w=dis[j];
			}
		}
		vis[p]=1;
		sum+=w;
		//cout<<sum<<"sum "<<endl;
		for(int j=0;j<n;j++)
		{
			if(!vis[j]&&(dis[j]==-1||dis[j]>map[p][j]))
			{
				dis[j]=map[p][j];
			}
		}
	}
	printf("%d\n",sum);
}
int main()
{
	while(scanf("%d",&n)==1)
	{
		for(int i=0;i<4;i++)
		{
			for(int j=0;j<4;j++)
			{
				scanf("%d",&map[i][j]);
				if(i!=j&&!map[i][j])
                	 map[i][j]=INF;//相对于agri-net,多了需要排除权值为0的情况
			}
		}
		Prim();
	}
	return 0;
 } 

50%ac代码（明明是对的QAQ）