题目1024:畅通工程
时间限制:1 秒
内存限制:32 兆
特殊判题:否
提交:5868
解决:2152
-
题目描述:
-
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
-
输入:
-
测试输入包含若干测试用例。每个测试用例的第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 ?
-
来源:
- 2007年浙江大学计算机及软件工程研究生机试真题
-
-
#include <cstdio> #include <algorithm> #define MAX 105 using namespace std; int tree[MAX]; int findRoot(int x){ if(tree[x]==-1)return x; else{ int temp=findRoot(tree[x]); tree[x]=temp; return temp; } } struct edge{ int node1,node2,length; bool operator <(const struct edge &b)const{ return length<b.length; } }edge[MAX]; int main(){ int n,m; while(scanf("%d%d",&n,&m)!=EOF&&n!=0){ for(int i=0;i<MAX;i++){ tree[i]=-1; } for(int i=0;i<n;i++){ scanf("%d%d%d",&edge[i].node1,&edge[i].node2,&edge[i].length); } sort(edge,edge+n); int ans=0; int a,b; for(int i=0;i<n;i++){ a=findRoot(edge[i].node1); b=findRoot(edge[i].node2); if(a!=b){ ans+=edge[i].length; tree[a]=b; } } int flag=0; for(int i=1;i<=m;i++){ if(tree[i]==-1)flag++; } if(flag==1){ printf("%d\n",ans); }else{ printf("?\n"); } } return 0; }
与前面几道MST的题,多了一个是否生成最小生成树的判断。