思路
最小生成树板子题,用Kruskal硬过就行了。
算法流程:
初始设点在集合A,无连边;
将图中的边按照权值从小到大排序,然后从最小的边开始连接(使用并查集);
将边的两端点加入集合B,如果当前边的两个端点不都在集合A,则进行连接;
选择m-1条边的时候刚好将m个点连上,此时判断权值之和是否大于c。
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn=10005,maxm=100005;
struct E{
int from,next,to,dis;
}edge[maxm*2];
int c,n,m,u,v,w;
int head[maxn],cnt=0,fa[maxn];
int find(int x){
if(fa[x]==x) return x;
else return fa[x]=find(fa[x]);
}
void unite(int x,int y){
fa[find(x)]=find(y);
}
void addedge(int from,int to,int dis){
edge[++cnt].next=head[from];
edge[cnt].from=from;
edge[cnt].to=to;
edge[cnt].dis=dis;
head[from]=cnt;
}
bool cmp(E a,E b){
return a.dis<b.dis;
}
int main(){
scanf("%d%d%d",&c,&n,&m);
for(int i=1;i<=n;i++){
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w);
addedge(v,u,w);
}
int tot=0,sm=0;
for(int i=1;i<=m;i++) fa[i]=i;
sort(edge+1,edge+1+cnt,cmp);
for(int i=1;i<=cnt;i++){
if(find(edge[i].to)!=find(edge[i].from)){
sm+=edge[i].dis;
unite(edge[i].to,edge[i].from);
tot++;
}
if(tot==m-1) break;
}
if(sm<=c) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
return 0;
} 
京公网安备 11010502036488号