洛谷P1195口袋的天空

#include
#include<string>
#include
#include
#include
#include
#include
#include
using namespace std;

int n,m,k;
int sum=0,ans=0;
int pre[1001];

struct bian {
    int u,v,w;
} b[10001];

bool comp(bian a,bian b){
    return a.w<b.w;
}
int find(int x) {
    if(pre[x]==x)return x;
    return pre[x]=find(pre[x]);
}

int main() {
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1; i<=n; i++) pre[i]=i;
    for(int i=1; i<=m; i++) {
        scanf("%d%d%d",&b[i].u,&b[i].v,&b[i].w);
    }
    stable_sort(b+1,b+m+1,comp);
    for(int i=1; i<=m; i++) {
        int fx=find(b[i].u),fy=find(b[i].v);
        if(fx!=fy) {
            pre[fx]=fy;
            sum++;
            ans+=b[i].w;
        }
        if(sum==n-k) {
            cout<<ans;
            return 0;
        }
    }
    printf("No Answer");
    return 0;
}