还是畅通工程 （最小生成树）

题目描述

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

输入格式

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( N< 100)；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。当N为0时，输入结束，该用例不被处理。

输出格式

对每个测试用例占用一行，输出最小的公路总长度。

Sample Input

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

Sample Output

3
5

题目分析

这道题是最基本的最小生成树模板题了。。。这里采用prim算法。

Code:

#include<iostream>
#include<cstring>
using namespace std;
#define INF 0x3f3f3f3f
#define maxn 1000
int map[maxn][maxn];
int d[maxn];
int visit[maxn];
int sum;
int minn; int temp;

int n, m; //n个顶点，m条边
int prim()
{
	visit[1] = false; //visit[i]=1表示可以访问该点，visit[i]=0表示不可以访问该点
	d[1] = 0;
	for (int i = 1; i < n; i++) //循环n-1次，找到n-1条边
	{
		minn = INF;
		temp = -1;
		for (int j = 1; j <= n; j++)  //循环n次，找到未加入最小生成树的最小权值的边和点
		{
			if (visit[j] && d[j] < minn)
			{
				minn = d[j];
				temp = j;
			}
		}
		if (temp == -1)
		{
			sum == INF;
			return 0;
		}
		visit[temp] = 0;
		sum += minn;  //更新最小生成树的权值
		for (int k = 1; k <= n; k++)  //更新未加入最小生成树的点和边
		{
			if (visit[k] && d[k] > map[k][temp])
			{
				d[k] = map[k][temp];
			}
		}
	}
	return 0;
}
int main()
{
	while (cin >> n && n)
	{
		memset(map, INF, sizeof(map)); //初始化每两点间的权值为无穷
		memset(d, INF, sizeof(d));  //初始化每个点到源点的距离为无穷
		memset(visit, 1, sizeof(visit)); //初始化每个点均可访问
		sum = 0;
		m = n * (n - 1) / 2;
		int x, y, w;//边（x，y）权值为w
		for (int i = 0; i < m; i++) //存图 ：m条边
		{
			cin >> x >> y >> w;
			map[x][y] = map[y][x] = w;
		}
		for (int i = 1; i <= n; i++) //初始化每个点到源点的距离（1是源点）
		{
			d[i] = map[i][1];
		}
		prim();
		if (sum == INF)
		{
			cout << "No" << endl;  //该图不连通，不能生成最小生成树
		}
		else
		{
			cout << sum << endl;
		}
	}
	return 0;
}