洛谷 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 }
