2018年全国多校算法寒假训练营练习比赛（第四场） - B - 道路建设（最小生成树）

链接：https://www.nowcoder.com/acm/contest/76/B
来源：牛客网

题目描述 
随着如今社会的不断变化，交通问题也变得越来越重要，所以市长决定建设一些公路来方便各个城市之间的贸易和交易。虽然市长的想法很好，但是他也遇到了一般人也经常头疼的问题，那就是手头的经费有限……在规划过程中，设计师们已经预算出部分城市之间建设公路的经费需求。现在市长想知道，它能不能将他的m个城市在有限的经费内实现公路交通。如果可以的话，输出Yes，否则输出No（两个城市不一定要直接的公路相连，间接公路到达也可以。）
输入描述:
测试输入包含多条测试数据
每个测试数据的第1行分别给出可用的经费c(<1000000)，道路数目n(n<10000)，以及城市数目m(<100)。
接下来的n行给出建立公路的成本信息，每行给出三个整数，分别是相连的两个城市v1、v2(0<v1,v2<=m)以及建设公路所需的成本h(h<100)。
输出描述:
对每个测试用例，输出Yes或No。
示例1
输入

20 10 5
1 2 6
1 3 3
1 4 4
1 5 5
2 3 7
2 4 7
2 5 8
3 4 6
3 5 9
4 5 2
输出

Yes
示例2
输入

10 2 2
1 2 5
1 2 15
输出

Yes
备注:
两个城市之间可能存在多条线路

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 1e4 + 10;
int n,m,c;
struct node{
    int u,v,cost;
}e[N];
bool cmp(node a,node b){
    return a.cost<b.cost;
}
int fa[110];
int find_fa(int x){
    int be = x;
    while(fa[x]!=x) x = fa[x];
    while(fa[be]!=be){
        int tem = fa[be];
        fa[be] = x;
        be = tem;
    }
    return x;
}
int main()
{
    while(scanf("%d%d%d",&c,&n,&m)!=EOF){
        for(int i=1;i<=m;i++) fa[i] = i;
        for(int i=0;i<n;i++) scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].cost);
        sort(e,e+n,cmp);
        int ans = 0;
        for(int i=0;i<n;i++){
            int faa = find_fa(e[i].u), fbb = find_fa(e[i].v);
            if(faa==fbb) continue;
            fa[faa] = fbb;
            ans += e[i].cost;
        }
        printf("%s\n",ans<=c?"Yes":"No");
    }
}