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

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

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 65536K，其他语言131072K
64bit IO Format: %lld

题目描述

随着如今社会的不断变化，交通问题也变得越来越重要，所以市长决定建设一些公路来方便各个城市之间的贸易和交易。虽然市长的想法很好，但是他也遇到了一般人也经常头疼的问题，那就是手头的经费有限……在规划过程中，设计师们已经预算出部分城市之间建设公路的经费需求。现在市长想知道，它能不能将他的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<cstring>
#include <iostream>  
#include<algorithm>
using namespace std;  
#define MAXV 105
#define inf 1<<29  
  
int map[MAXV][MAXV],n,c,m; 
  
void prim(){  
    int i,j,d[MAXV],vis[MAXV],mi,v;  
    for(i=1;i<=n;i++){  
        d[i]=map[1][i];  
        vis[i]=0;  
    }  
    for(i=1;i<=n;i++){  
        mi=inf;  
        for(j=1;j<=n;j++)  
            if(!vis[j] && d[j]<mi){  
                mi=d[j];  
                v=j;  
            }  
        vis[v]=1;  
        for(j=1;j<=n;j++)  
            if(!vis[j] && d[j]>map[v][j])  
                d[j]=map[v][j];  
    }  
    for(d[0]=0,i=1;i<=n;i++) d[0]+=d[i];  
  
    //printf("%d\n",d[0]);  
    if(d[0]>c)puts("No");
    else puts("Yes");
}  
  
int main(){    
    scanf("%d%d%d",&c,&m,&n);        
    int x,y,d;
    memset(map,0,sizeof(map));
    for(int i=0;i<m;i++){
    	scanf("%d%d%d",&x,&y,&d);
    	if(map[x][y]>0)d=min(d,map[x][y]);
    	map[x][y]=map[y][x]=d;
	}
        prim();  
    
    return 0;  
}