vijos P1234口袋的天空(Kruskal)(最小生成树)

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <climits>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <list>
#include<functional>
#define mod 1000000007
#define inf 0x3f3f3f3f
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int N=1005;

struct Edg
{
    int v,u;int w;
}edg[300*300+50];
bool cmp(Edg g,Edg h)
{
    return g.w<h.w;
}
int n,m,k,cnt=1,maxn;
int parent[N];
void init()
{
    for(int i=0;i<n;i++)parent[i]=i;
}
void Build()
{
    int u,v,w;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        edg[i].u=u;edg[i].v=v;edg[i].w=w;
    }
    if(n<k){printf("No Answer\n");exit(0);}
    sort(edg,edg+m,cmp);
}
int Find(int x) {
    if(parent[x] != x) parent[x] = Find(parent[x]);
    return parent[x];
}//查找并返回节点x所属集合的根节点
void Union(int x,int y) {
    x = Find(x);
    y = Find(y);
    if(x == y) return;
    parent[y] = x;
}//将两个不同集合的元素进行合并
void Kruskal()
{
    int sum=0;
    int num=0;
    int u,v;
    for(int i=0;i<m;i++)
    {
        u=edg[i].u;v=edg[i].v;
        if(Find(u)!=Find(v))
        {
            sum+=edg[i].w;

            num++;
            Union(u,v);
        }
        if(num>=n-(k-1)-1){
             printf("%d\n",sum);
             break;
        }
    }
    if(num<n-(k-1)-1)printf("No Answer\n");
}
int main() {
        scanf("%d%d%d",&n,&m,&k);
        init();
        Build();
        Kruskal();
    return 0;
}