P1195 口袋的天空

算法

最小生成树+kruskal

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 1005
using namespace std;
int n,m,k,x,y,l,sum,ans;
int fa[N];
struct Edge
{
    int u,v,w;
    bool operator <(Edge a) const    //重置运算符的函数写在里面的话必须加const
    {
        return w<a.w;
    }
}edge[N*10];
/*bool operator <(Edge a,Edge b)    //重置运算符的函数写在外面的话不能加const
{
        return a.w<b.w;
}*/
int find(int x)
{
    return fa[x]==x?fa[x]:fa[x]=find(fa[x]);
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
    }
    sort(edge+1,edge+m+1);    
    for(int i=1;i<=m;i++)
    {
        int fx=find(edge[i].u),fy=find(edge[i].v);
        if(fx!=fy)        
        {
            fa[fx]=fy;
            sum++;        
            ans+=edge[i].w;  
        }
        if(sum==n-k)        
        {
            printf("%d",ans);
            return 0;
        }
    }
    puts("No Answer");  
    return 0;
}