题目背景 Background
小杉坐在教室里,透过口袋一样的窗户看口袋一样的天空。 有很多云飘在那里,看起来很漂亮,小杉想摘下那样美的几朵云,做成棉花糖。
题目描述 Description
给你云朵的个数N,再给你M个关系,表示哪些云朵可以连在一起。
现在小杉要把所有云朵连成K个棉花糖,一个棉花糖最少要用掉一朵云,小杉想知道他怎么连,花费的代价最小。
输入输出格式 Input/output
输入格式:
每组测试数据的
第一行有三个数N,M,K(1<=N<=1000,1<=M<=10000,1<=K<=10)
接下来M个数每行三个数X,Y,L,表示X云和Y云可以通过L的代价连在一起。(1<=X,Y<=N,0<=L<10000)
30%的数据N<=100,M<=1000
输出格式:
对每组数据输出一行,仅有一个整数,表示最小的代价。
如果怎么连都连不出K个棉花糖,请输出’No Answer’。
输入样例:
3 1 2
1 2 1
输出样例:
1
大体意思是给一个图,生成K个森林,可以通过稍稍改变Kruskal算法来实现生成,并查集初始化后是有N个森林,每次合并两个集合只会令森林数减一,生成K个森林只需要合并N-K次即可。
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int size = 200010;
struct dc
{
int f,t,d;
}l[size];
int f[size];
int find(int x)
{
if(f[x] == x)
return x;
return f[x] = find(f[x]);
}
bool cmp(dc a,dc b)
{
return a.d < b.d;
}
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
k = n-k;
int ans = 0;
for(int i = 1 ; i <= m ; i ++)
scanf("%d%d%d",&l[i].f,&l[i].t,&l[i].d);
sort(l+1,l+m+1,cmp);
for(int i = 1 ; i <= n ; f[i] = i , i ++);
for(int i = 1 ; i <= m ; i ++)
{
int x = find(l[i].f);
int y = find(l[i].t);
if(x != y)
{
k --;
f[x] = y;
ans += l[i].d;
}
if(k == 0)
break;
}
if(k)
puts("No Answer");
else
cout<<ans;
return 0;
}
传送门:洛谷 1995 口袋的天空