1004: 惠民工程 （2013年中南大学研究生复试机试 ）

1004: 惠民工程

时间限制: 1 Sec  内存限制: 128 MB
提交: 404  解决: 81
[提交] [状态] [讨论版] [命题人:外部导入]

题目描述

市政府“惠民工程”的目标是在全市n个居民点间之架设煤气管道（但不一定有直接的管道相连，只要能间接通过管道可达即可）。很显然最多可架设 n(n-1)/2条管道,然而实际上要连通n个居民点只需架设n-1条管道就可以了。现请你编写程序，计算出该惠民工程需要的最低成本。

输入

测试输入包含若干测试用例。每个测试用例的第1行给出居民点数目M ( < =100 )、 评估的管道条数 N；随后的 N 行对应居民点间管道的成本，每行给出一对正整数，分别是两个居民点的编号，以及此两居民点间管道的成本（也是正整数）。为简单起见，居民点从1到M编号。

输出

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

样例输入

3 3
1 2 1
1 3 2
2 3 4
3 1
2 3 2

样例输出

3
?

来源/分类

2013年中南大学研究生复试机试  

 

 

#include<iostream>
#include<stdio.h> 
#include<algorithm>
using namespace std;
/*要畅通，就必需要m-1条边，要使费用最小则尽量用权值最小的变，先对所有的边进行排序
然后从小到大依次选择，判断该边是否可以加入此工程,判断的方法是使用并查集*/
int n, m;
int arr[105];
struct node
{
    int start, end, w;
}a[5010];

int cmpare(node x, node y)
{
    return x.w<y.w;
}

int find(int x){
    while(arr[x]!=x){
        x=arr[x];
    }
    return x;
}

int add(int x,int y){
    int x1=find(x);
    int y1=find(y);
    if(x1!=y1){
        arr[x1]=y1;
        return 1;
    }
    return 0;
}
int main(){
    while (scanf("%d%d", &m, &n) != EOF){
        for (int i = 0; i < n; i++){
            scanf("%d%d%d", &a[i].start, &a[i].end, &a[i].w);
        }
        sort(a, a + n, cmpare);
        for (int i = 1; i <=m; i++){
            arr[i] = i;
        }
        int num = 0;
        int result = 0;
        for (int i = 0; num < m - 1 && i < n; i++){
            if (add(a[i].start,a[i].end)){
                num++;
                result += a[i].w;
            }
        }
        if (num == m - 1){
            printf("%d\n", result);
        }
        else{
            printf("?\n");
        }
    }
}