【从0开始的刷题生活】PTA 数据结构2019-秋 08-图7 公路村村通 (30 分)

题目

原题地址
现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:
输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。

输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−1，表示需要建设更多公路。

输入样例:
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

题目分析

阅读题目，题目要求输出最小成本，可以看出其为求一个最小生成树，在最小生成树的两个算法Prim算法和Kruskal算法中，此题可以选择Prim算法

代码

#include <iostream>
#define maxn 1005
#define infinity 30000
using namespace std;
typedef struct {
	int nv,ne;
	int arcs[maxn][maxn]; 
}MGraph;
typedef struct{
	int v1,v2;
	int value;
}Edge,*enode;
int FindMinDist(MGraph g,int dist[])//找到最小
{
	int minv,v;
	int mindist=infinity;
	for(v=0;v<g.nv;v++){
		if(dist[v]!=0 && dist[v]<mindist){
			mindist=dist[v];
			minv=v;
		}
	}
	if(mindist<infinity) return minv;
	else return -1;
}
int Prim(MGraph g )
{
	int dist[maxn],totalweight;
	int parent[maxn],v,w;
	int vcount;
	enode e;
	for(v=0;v<g.nv;v++)
	{
		dist[v]=g.arcs[0][v];
		parent[v]=0;
	}
	totalweight=0;
	vcount=0;
	e=(enode)malloc(sizeof(Edge));
	dist[0]=0;
	vcount++;
	parent[0]=-1;
	while(1)
	{
		v=FindMinDist(g,dist);
		if(v==-1) break;
		e->v1=parent[v];
		e->v2=v;
		e->value=dist[v];
		totalweight+=dist[v];
		dist[v]=0;
		vcount++;
		for(w=0;w<g.nv;w++)
			if(dist[w]!=0 && g.arcs[v][w]<infinity){
				if(g.arcs[v][w]<dist[w]){
					dist[w]=g.arcs[v][w];
					parent[w]=v;
				}
			}
	}
	if(vcount<g.nv) totalweight=-1;
	return totalweight;
}
int main()
{
	int n,m,a,b,c;
	MGraph g;
	cin>>n>>m;
	g.nv=n;
	g.ne=m;
    for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			g.arcs[i][j]=infinity;
	for(int i=0;i<m;i++)
	{
		cin>>a>>b>>c;
		g.arcs[a-1][b-1]=c;
		g.arcs[b-1][a-1]=c;
	}
	cout<<Prim(g)<<endl;
}