P1195 口袋的天空

题目

P1195 口袋的天空_第1张图片

思路

一眼最小生成树
只需要简单存图然后并查集模拟连接云朵(只需要连接n-k个云朵,因为需要将若干个云朵变成K个云朵)

代码

#include
using namespace std;
int n,m,k;
int p[1005];
int ans;
struct Edge{int u,v,w;}edges[10005];//存边
struct cmp { bool operator () (const Edge &a,const Edge &b) { return a.w<b.w; } };//给边排序
inline int find(int x) { p[x]==x?x:p[x]=find(p[x]); }//并查集 
inline void init(int n) { for(int i=1;i<=n;i++) p[i]=i; }//并查集初始化
int main()
{
	cin>>n>>m>>k;
	for(int i=1;i<=m;i++) cin>>edges[i].u>>edges[i].v>>edges[i].w;
	init(n);
	sort(edges+1,edges+1+m,cmp());
	int cnt=0;
	for(int i=1;i<=m;i++){
		Edge &e=edges[i];int x=find(e.u),y=find(e.v);
		if(x!=y) p[x]=y,ans+=e.w,cnt++;//模拟连接边
		if(k==n-cnt) { cout<<ans;return 0; }
	}
	cout<<"No Answer";//若循环结束还没有剩下k个就无解
	return 0;
}

你可能感兴趣的:(c++,数据结构,算法,最小生成树,Kruskal)