题目描述
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
输入
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M (N, M < =100 );随后的 N 行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
输出
对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。
样例
输入:
3 4
1 2 1
2 3 2
3 4 3
2 4
1 2 1
3 4 2
0 5
输出:
6
?
解题思路:
使用Prim算法求解最小生成树
下面为AC代码:
#include <iostream>
#include <algorithm>
#include<string>
using namespace std;
const int prim_max_num = 110;
const int INF = 1000000000;
int G[prim_max_num][prim_max_num];
int dis[prim_max_num]; //顶点与集合S的最短距离
bool visited[prim_max_num] = {false};
int prim(int n)
{
//默认根结点为 0
fill(dis, dis + prim_max_num, INF); //初始化 dis 数组
dis[1] = 0;
int result = 0;
for (int i = 1; i <= n; i++)
{ //共有 n 个结点
int index = -1; //最短路径结点的下标
int min = INF;
for (int j = 1; j <= n; j++)
{
if (visited[j] == false && dis[j] < min)
{
index = j;
min = dis[j];
}
}
//找不到最小结点
if (index == -1)
{
return -1;
}
visited[index] = true;
result += dis[index];
for (int k = 1; k <= n; k++)
{
if (visited[k] == false && G[index][k] != INF && G[index][k] < dis[k])
{
dis[k] = G[index][k];
}
}
}
return result;
}
int main()
{
int n,m;
while(cin>>n>>m){
if(n == 0){
break;
}
fill(G[0], G[0] + prim_max_num * prim_max_num, INF);
fill(visited,visited + prim_max_num,false);
int from_node, to_node, weight;
for (int i = 0; i < n; i++)
{
cin >> from_node >> to_node >> weight;
G[from_node][to_node] = G[to_node][from_node] = weight;
}
int result = prim(m);
if(result != -1){
cout << result << endl;
}else{
cout << "?" << endl;
}
}
system("pause");
return 0;
}