道路建设

2018年全国多校算法寒假训练营练习比赛（第四场）

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

题目描述

随着如今社会的不断变化，交通问题也变得越来越重要，所以市长决定建设一些公路来方便各个城市之间的贸易和交易。虽然市长的想法很好，但是他也遇到了一般人也经常头疼的问题，那就是手头的经费有限……在规划过程中，设计师们已经预算出部分城市之间建设公路的经费需求。现在市长想知道，它能不能将他的m个城市在有限的经费内实现公路交通。如果可以的话，输出Yes，否则输出No（两个城市不一定要直接的公路相连，间接公路到达也可以。）

输入描述:

测试输入包含多条测试数据
每个测试数据的第1行分别给出可用的经费c(<1000000)，道路数目n(n<10000)，以及城市数目m(<100)。
接下来的n行给出建立公路的成本信息，每行给出三个整数，分别是相连的两个城市v1、v2(0 
   
输出描述:
 
   
对每个测试用例，输出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 

using namespace std;

const int N = 2e5 + 5;
int n, m, C, fa[N];

struct node
{
    int u, v, c;
} a[N];

int find(int x)
{
    return x == fa[x] ? x : fa[x] = find(fa[x]);
}

bool cmp(node a, node b)
{
    return a.c < b.c;
}

int main()
{
    int n, m;

    while(cin >> C >> m >> n)
    {
        for(int i = 1; i <= n; i++)
            fa[i] = i;

        for(int i = 0; i < m; i++)
            scanf("%d%d%d", &a[i].u, &a[i].v, &a[i].c);

        sort(a, a + m, cmp);
        long long res = 0;

        for(int i = 0; i < m; i++)
        {
            int u = find(a[i].u), v = find(a[i].v);

            if(u == v)
                continue;

            fa[u] = fa[v];
            res += a[i].c;
        }

        puts(res <= C ? "Yes" : "No");
    }

    return 0;

}