口袋的天空-最小生成树

http://www.luogu.org/problem/show?pid=1195

题目背景

小杉坐在教室里，透过口袋一样的窗户看口袋一样的天空。

有很多云飘在那里，看起来很漂亮，小杉想摘下那样美的几朵云，做成棉花糖。

题目描述

给你云朵的个数N，再给你M个关系，表示哪些云朵可以连在一起。

现在小杉要把所有云朵连成K个棉花糖，一个棉花糖最少要用掉一朵云，小杉想知道他怎么连，花费的代价最小。

输入输出格式

输入格式：

每组测试数据的

第一行有三个数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‘。

 1 #include<iostream>
 2 #include<vector>
 3 #include<algorithm>
 4 using namespace std;
 5 const int maxn=1010;
 6 struct Edge{
 7     int from,to,dist;
 8     bool operator<(const Edge& b)const{
 9         return dist<b.dist;
10     }
11 };
12 int fa[maxn],m,n,k;
13 vector<Edge> edges;
14 int init()
15 {
16     for(int i=1;i<=n;i++) fa[i]=i;
17 }
18 int find(int x)
19 {
20     return fa[x]==x?x:fa[x]=find(fa[x]);
21 }
22 int main()
23 {
24     cin>>n>>m>>k;
25     init();
26     for(int i=1;i<=m;i++){
27         int x,y,z;cin>>x>>y>>z;
28         edges.push_back((Edge){x,y,z});
29     }
30     sort(edges.begin(),edges.end());
31     int ans=0;k=n-k;
32     for(int i=0;i<m;i++){
33         int x,y,z;
34         x=edges[i].from,y=edges[i].to,z=edges[i].dist;
35         if(find(x)!=find(y)){
36             k--;
37             ans+=z;
38             fa[find(x)]=find(y);
39             if(!k) break;
40         }
41     }
42     if(k) cout<<"No Answer";
43     else cout<<ans;
44     return 0;
45 }