【洛谷】P1195 口袋的天空（k个生成树+kruskal）

 

解题思路：一开始没看懂题目，把题目描述转化一下其实题目让我们求的是k个生成树， 那么这该怎么做呢，我们思考一下kruskal算法的原理，一开始图中的每个点都是一棵树，我们查询两个点是否在同一个树中，如果不在，则将两个数合并成一个树，这样问题是不是就迎刃而解了，我们只需要用一个变量cur存储当前图中有几个树，每次合并树的时候cur减1，当cur的值等于k的时候，合并树所需要的花费也就是最后的答案

下面附上ac代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <string>
#include <cstring>
#include <queue>
#include <set>
#include <map>
#include <vector>
#define dbg(a)  cout<<#a<<" : "<<a<<endl;
#define IOS std::ios::sync_with_stdio(false);cin.tie(0),cout.tie(0);
#define PAUSE   system("pause")
#define sd(a)       scanf("%d",&a)
#define sll(a)      scanf("%intd",&a)
#define sdd(a,b)    scanf("%d%d",&a,&b)
#define sddd(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define sf(a)       scanf("%lf",&a)
#define sff(a,b)    scanf("%lf%lf",&a,&b)
#define sfff(a,b,c) scanf("%lf%lf%lf",&a,&b,&c)
using namespace std;
typedef long long ll;
typedef pair<int,int>  pii;
const int maxn=1e4+5;
int n,m,maxx,k,cnt,ans,idx,cur;
//vector<pii> ve[maxn];
int fa[maxn];
struct Edge{
    int u,v,w;
}edge[maxn];
bool cmp(Edge a,Edge b)
{
    return a.w<b.w;
}
void init(int n)
{
    for(int i=1;i<=n;i++)
        fa[i]=i;
}
int find(int x)
{
    return x==fa[x]?x:(fa[x]=find(fa[x]));
}
void kruskal()
{
    for(int i=1;i<=m;i++)
    {
        int x=find(edge[i].u);
        int y=find(edge[i].v);
        int z=edge[i].w;
        if(x==y) continue;
        else{
            fa[x]=y;
            cur--;
            maxx+=z;
            if(cur==k)  return;
        }
    }
}
int main()
{
    sddd(n,m,k);
    if(k>n) {
        printf("No Answer\n");
        return 0;
    }
    init(n);
    cur=n;
    for(int i=1;i<=m;i++)
    {
        sddd(edge[i].u,edge[i].v,edge[i].w);
    }
    sort(edge+1,edge+m+1,cmp);
    kruskal();
    printf("%d\n",maxx);
    //PAUSE;
    return 0;
}