题目描述
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
输入
测试输入包含若干测试用例。每个测试用例的第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
?
最小生成树的模板改动,也可以用并查集来写。
代码:
# include <stdio.h>
# include <string.h>
# include <algorithm>
using namespace std;
int main(void)
{
int n, m, book[101], a[101][101], dis[101];
while (~ scanf("%d %d", &n, &m))
{
if (n == 0)
break;
int i, j, k;
memset(book, 0, sizeof(book));
for (i = 1; i <= m; i ++)
for (j = 1; j <= m; j ++)
if (i == j)
a[i][j] = 0;
else
a[i][j] = 99999999;
int x, y, v;
for (i = 1; i <= n; i ++)
{
scanf("%d %d %d", &x, &y, &v);
a[x][y] = a[y][x] = v;
}
for (i = 1; i <= m; i ++)
dis[i] = a[1][i];
book[1] = 1;
int count = 1, minn, sum = 0;
while (count < m) // 最小生成树模板
{
minn = 99999999;
for (i = 1; i <= m; i ++)
{
if (book[i] == 0 && dis[i] < minn)
{
minn = dis[i];
j = i;
}
}
book[j] = 1;
count ++;
sum += dis[j];
for (k = 1; k <= m; k ++)
{
if (book[k] == 0 && dis[k] > a[j][k])
dis[k] = a[j][k];
}
}
int qwe = 0;
for (i = 1; i <= m; i ++) // 判断是不是能构成通路。
{
if (book[i] == 0)
{
qwe = 1;
break;
}
}
if (qwe == 0)
printf("%d\n", sum);
else
printf("?\n");
}
return 0;
}