P1195 口袋的天空 (生成树)

题目链接

解法：

初始n个节点，n颗树，每连一条边，减少一棵树。k棵树需要连n-k条边。。。1棵树需要连n-1条边。。。

给每条可以连的边按代价从小到大排个序，然后连n-k条边造k个最小生成树就可以了。

Code:

#include <bits/stdc++.h>
# define LL long long
using namespace std;

const int maxn=1000+10;
const int maxm=10000+10;
int parent[maxn];
int N, M, K;

struct Edge{
    int u,v,w;
}e[maxm];

int find(int p){
    if(p==parent[p]) return p;
    parent[p]=find(parent[p]);
    return parent[p];
}

int main(){
    scanf("%d %d %d", &N, &M, &K);
    for(int i=1;i<=N;++i){
        parent[i]=i;
    }
    for(int i=1;i<=M;++i){
        int u,v,w;
        scanf("%d %d %d", &e[i].u, &e[i].v, &e[i].w);
    }
    sort(e+1,e+M+1,[](Edge e1, Edge e2){
        return e1.w<e2.w;
    });
    int res=0;
    int cnt=0;
    for(int i=1;i<=M;++i){
        int u=e[i].u;
        int v=e[i].v;
        if(find(u)==find(v)) continue;
        cnt++;
        res+=e[i].w;
        parent[find(u)]=find(v);
        if(cnt==N-K){
            printf("%d", res);
            return 0;
        }
    }
    printf("No Answer");
    return 0;
}