刷题记录:牛客NC15108道路建设

传送门:牛客

题目描述:

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

一道最小生成树的板题.不会的可以移步到这篇博客学习,详细解释了$kruskal,prim$算法

对于这道题,重边似乎比较多,但是由于边的范围又不是很大,所以我们还是可以直接套模板的,不然可能需要精简一下我们的边再跑最小生成树

那么这篇博客就不在赘述了,本题博主使用了$kruskal$算法(因为比较好打)

下面是具体的代码部分:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
#include <deque>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 10010
#define ll_maxn 0x3f3f3f3f3f3f3f3f
const double eps=1e-8;
struct Node{
	int u,v,w;
}edge[maxn];
int fa[maxn];
int find(int x) {
	if(x==fa[x]) return x;
	return fa[x]=find(fa[x]);
}
int c,n,m;
bool cmp(Node a,Node b) {
	return a.w<b.w;
}
int main() {
	while(~scanf("%d%d%d",&c,&m,&n)) {
		memset(edge,0,sizeof(edge));
		for(int i=1;i<=n;i++) fa[i]=i;
		for(int i=1;i<=m;i++) {
			edge[i].u=read();edge[i].v=read();edge[i].w=read();
		}
		sort(edge+1,edge+m+1,cmp);
		int ans=0;int cnt=0;
		for(int i=1;i<=m;i++) {
			int x=find(edge[i].u),y=find(edge[i].v);
			if(x==y) continue;
			fa[x]=y;
			ans+=edge[i].w;
			cnt++;
			if(cnt==n-1) break;
		}
		if(ans<=c) puts("Yes");
		else puts("No");
	}
	return 0; 
}