洛谷1195 修复公路 最小生成树

题意：求生成k棵树的最小总路程

题解：显然是最小生成树，但是关键是k棵。其实一开始可以看成是N棵树，没加入一条边，只要不生成环，就减少了一棵树，所以只要不成环的前提下加入n-k条边就生成了k棵树。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
struct edge{
    int u,v,cost;
};
int cmp(edge a,edge b){
    return a.cost<b.cost;
}
edge e[10005];
int V,E,K;
int p[1005];
void in_it(int n){
    int i;
    for(i=1;i<=n;i++)p[i]=i;
}
int finde(int a){
    if(a==p[a])return a;
    return p[a]=finde(p[a]);
}
void join(int a,int b){
    a=finde(a);
    b=finde(b);
    p[a]=b;
}
bool same(int a,int b){
    a=finde(a);
    b=finde(b);
    if(a==b)return true;
    return false;
}
int kruskal(){
    int cnt=0;
    sort(e,e+E,cmp);
    in_it(V);
    int ans=0,i;
    for(i=0;i<E;i++){
        edge tem=e[i];
        if(same(tem.u,tem.v))continue;
        cnt++;
        join(tem.u,tem.v);
        ans+=tem.cost;
        if(cnt==V-K)return ans;
    }
    return -1;
}
int main()
{
    int i,x,y,l,tot=0;
    scanf("%d%d%d",&V,&E,&K);
    for(i=0;i<E;i++){
        scanf("%d%d%d",&x,&y,&l);
        e[tot].u=x;
        e[tot].v=y;
        e[tot].cost=l;
        tot++;
    }
    int ans=kruskal();
    if(ans==-1)printf("NO Answer\n");
    else printf("%d\n",ans);
    return 0;
}