洛谷 1195 口袋的天空

【题解】

　　最小生成树。区别仅仅是不需要连城一棵树，而是连成k棵树，也就是选择最小生成树里面最小的k条边。

　

 1 #include<cstdio>
 2 #include<algorithm>
 3 #include<cstring>
 4 #define LL long long
 5 #define rg register
 6 #define N 200010
 7 using namespace std;
 8 int n,m,k,cnt,f[N];
 9 struct edge{
10     int u,v,dis;
11 }a[N];
12 LL ans=0;
13 bool flag=0;
14 inline int read(){
15     int k=0,f=1; char c=getchar();
16     while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
17     while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
18     return k*f;
19 }
20 bool cmp(edge a,edge b){return a.dis<b.dis;}
21 int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
22 int main(){
23     n=read(); m=read(); k=read();
24     for(rg int i=1;i<=n;i++) f[i]=i;
25     for(rg int i=1;i<=m;i++) a[i].u=read(),a[i].v=read(),a[i].dis=read();
26     sort(a+1,a+1+m,cmp);
27     for(rg int i=1;i<=m;i++){
28         int u=find(a[i].u),v=find(a[i].v);
29         if(u!=v){
30             ans+=a[i].dis; f[u]=v; cnt++;
31         }
32         if(cnt>=n-k){
33             flag=1;
34             break;
35         }
36     }
37     if(flag) printf("%lld\n",ans);
38     else puts("No Answer");
39     return 0;
40 }

View Code

 

转载于:https://www.cnblogs.com/DriverLao/p/9398051.html