P2121 拆地毯-数论+Kruskal

拆地毯

题目背景

还记得 NOIP 2011 提高组 Day1 中的铺地毯吗？时光飞逝，光阴荏苒，三年过去了。组织者精心准备的颁奖典礼早已结束，留下的则是被人们踩过的地毯。请你来解决类似于铺地毯的另一个问题。

题目描述

会场上有 n 个关键区域，不同的关键区域由 m 条无向地毯彼此连接。每条地毯可由三个整数 u、v、w 表示，其中 u 和 v 为地毯连接的两个关键区域编号，w 为这条地毯的美丽度。

由于颁奖典礼已经结束，铺过的地毯不得不拆除。为了贯彻勤俭节约的原则，组织者被要求只能保留 K 条地毯，且保留的地毯构成的图中，任意可互相到达的两点间只能有一种方式互相到达。换言之，组织者要求新图中不能有环。现在组织者求助你，想请你帮忙算出这 K 条地毯的美丽度之和最大为多少。

输入格式

第一行包含三个正整数 n、m、K。

接下来 m 行中每行包含三个正整数 u、v、w。

输出格式

只包含一个正整数，表示这 K 条地毯的美丽度之和的最大值。

样例 #1

样例输入 #1

5 4 3
1 2 10
1 3 9
2 3 7
4 5 3

样例输出 #1

22

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define de(x) cout<<x<<" ";
#define sf(x) scanf("%d",&x);
#define Pu puts("");
const int N=1e5+10,M=2e5+10;
struct E{
    int x,y,u;
};
int n,m,k;
int ans;
int f[N];
struct cmp{
    bool operator()(E a,E b){
        return a.u<b.u;
    }
};
priority_queue<E,vector<E>,cmp>q;
int find(int x){
    if(x==f[x]) return x;
    return f[x]=find(f[x]);
}
int main(){
    cin>>n>>m>>k;
    int x,y,z;
    for(int i=1;i<=m;i++){
        sf(x)sf(y)sf(z)
        q.push(E{x,y,z});
    }
    int a,b;
    for(int i=1;i<=n;i++) f[i]=i;
    while(k){
        x=q.top().x;y=q.top().y;z=q.top().u;q.pop();
        a=find(x);b=find(y);
        if(a!=b){
            k--;
            if(a>b) f[a]=b;
            else f[b]=a;
            ans+=z;
        }
    }
    printf("%d\n",ans);
    return 0;
}