您的位置:首页 > 理论基础

题目1024:畅通工程 2007年浙江大学计算机及软件工程研究生机试真题

2014-03-09 22:38 411 查看
题目描述:
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。

输入:
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M (N, M < =100 );随后的 N 行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。

输出:
对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。

样例输入:
3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100


样例输出:
3
?

AC代码:

#include<stdio.h>
#include<algorithm>
using namespace std;

int p[101];

struct Road {
int start;
int end;
int cost;
bool operator < (const Road &b) const {
return cost < b.cost;
}
}roads[100];

int findRoot(int x) {
if(p[x]==-1)
return x;
else {
int tmp = findRoot(p[x]);
p[x] = tmp;
return tmp;
}
}

int Kruscal(Road r[], int n, int m) {
int a, b, j, t=0, sum=0;
sort(r,r+n);
for(j=0; j<n; j++) {
a = findRoot(r[j].start);
b = findRoot(r[j].end);
if(a!=b) {
p[b] = a;
sum += r[j].cost;
t++;
}
}
if(t == m-1)
return sum;
else
return -1;
}

int main() {
//freopen("in.txt","r",stdin);
int n, m;
while(scanf("%d %d",&n,&m)!=EOF && n) {
int i;
for(i=1; i<=m; i++) {
p[i] = -1;
}
for(i=0; i<n; i++) {
scanf("%d %d %d",&roads[i].start,&roads[i].end,&roads[i].cost);
}
int minCost = Kruscal(roads,n,m);
if(minCost >= 0)
printf("%d\n",minCost);
else
printf("?\n");
}

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐