nowcoder 道路建设

链接：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

#include<bits/stdc++.h>
using namespace std;
#define maxn 100000
int fa[maxn];
struct ac{
   int x,y,z;
   ac(){}
   ac(int a,int b,int c){
      x=a,y=b,z=c;
   }
   bool operator <(const ac &a) const{
       return z<a.z;
   }
}a[maxn];
int fin(int x){
  return fa[x]==x? x:fa[x]=fin(fa[x]);
}
int main(){
   int c,n,m;
   while(cin>>c>>n>>m){
     for(int j=0;j<n;j++){
        scanf("%d%d%d",&a[j].x,&a[j].y,&a[j].z);
     }
     for(int j=0;j<=m;j++){
       fa[j]=j;
     }
     sort(a,a+n);
     long long ans=0;
     int k=0;
     for(int j=0;j<n;j++){
        int u=fin(a[j].x);
        int v=fin(a[j].y);
        if(u!=v){
           fa[v]=u;
           ans+=a[j].z;
           k++;
        }
        if(k==m-1) break;
     }
     if(ans<=c){
        cout<<"Yes"<<endl;
     }
     else{
        cout<<"No"<<endl;
     }
   }
}