口袋的天空
题目背景
小杉坐在教室里,透过口袋一样的窗户看口袋一样的天空。
有很多云飘在那里,看起来很漂亮,小杉想摘下那样美的几朵云,做成棉花糖。
题目描述
给你云朵的个数 N N N,再给你 M M M 个关系,表示哪些云朵可以连在一起。
现在小杉要把所有云朵连成 K K K 个棉花糖,一个棉花糖最少要用掉一朵云,小杉想知道他怎么连,花费的代价最小。
输入格式
第一行有三个数 N , M , K N,M,K N,M,K。
接下来 M M M 行每行三个数 X , Y , L X,Y,L X,Y,L,表示 X X X 云和 Y Y Y 云可以通过 L L L 的代价连在一起。
输出格式
对每组数据输出一行,仅有一个整数,表示最小的代价。
如果怎么连都连不出
K
K
K 个棉花糖,请输出 No Answer
。
样例 #1
样例输入 #1
3 1 2
1 2 1
样例输出 #1
1
提示
对于 30 % 30\% 30% 的数据, 1 ≤ N ≤ 100 1 \le N \le 100 1≤N≤100, 1 ≤ M ≤ 1 0 3 1\le M \le 10^3 1≤M≤103;
对于 100 % 100\% 100% 的数据, 1 ≤ N ≤ 1 0 3 1 \le N \le 10^3 1≤N≤103, 1 ≤ M ≤ 1 0 4 1 \le M \le 10^4 1≤M≤104, 1 ≤ K ≤ 10 1 \le K \le 10 1≤K≤10, 1 ≤ X , Y ≤ N 1 \le X,Y \le N 1≤X,Y≤N, 0 ≤ L < 1 0 4 0 \le L<10^4 0≤L<104。
题意及分析
给出n个点,m个关系,把这些点连成k棵树,每棵树至少由一个点组成,问怎么连使总的代价最小。
很明显的最小生成树问题,经典的最小生成树问题是:给n个点,m条边,把n个点连成一棵树,问怎么连是总的代价最小,当然最小生成树也可能不存在。
而这道题又是恰好是Kruskal算法的变式。
Kruskal算法流程:
将m条边按权值从小到大排序后,枚举每一条边,如果该边不在连通块中,则加入连通块,直到选了n-1条边为止。
我们知道一棵n个点的树有n-1条边,那么在这道题中,我们把一棵树分成k棵树就需要去掉k-1条边,即需要连:n-1-(k-1) = n-k条边。
放代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e4+5;
const int INF=0x3f3f3f3f;
int n,m,k;
int map[N][N];
int f[N];
bool st[N];
struct edge{
int u,v,w;
bool operator < (const edge &W) const{
return w<W.w;
};
}edges[N];
int find(int x){
if(f[x]!=x) f[x]=find(f[x]);
return f[x];
}
int main(){
scanf("%d %d %d",&n,&m,&k);
if(k>n){
printf("No Answer\n");
return 0;
}
int u,v,w;
for(int i=0;i<m;++i){
scanf("%d %d %d",&u,&v,&w);
edges[i].u=u,edges[i].v=v,edges[i].w=w;
}
sort(edges,edges+m);
for(int i=1;i<=n;++i) f[i]=i;
int cnt=0,sum=0;
for(int i=0;i<m;++i){
u=edges[i].u,v=edges[i].v,w=edges[i].w;
int f_u=find(u);
int f_v=find(v);
if(f_u!=f_v){
f[f_u]=f_v;
++cnt;
sum+=w;
}
if(cnt>=n-k){
printf("%d",sum);
return 0;
}
}
printf("No Answer");
return 0;
}