首页 > 代码库 > 口袋的天空-最小生成树

口袋的天空-最小生成树

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 }

 

口袋的天空-最小生成树