洛谷 P2121 拆地毯

 题目链接：拆地毯 - 洛谷

题目背景

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

题目描述

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

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

输入格式

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

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

输出格式

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

输入输出样例

输入 #1

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

输出 #1

22

说明/提示

选择第 1、2、4 条地毯，美丽度之和为 10 + 9 + 3 = 22。

若选择第 1、2、3 条地毯，虽然美丽度之和可以达到 10 + 9 + 7 = 26，但这将导致关键区域 1、2、3 构成一个环，这是题目中不允许的。

1<=n,m,k<=100000

涉及算法：并查集，排序

思路：

先把每条边根据美丽度大小从大到小排序，把每个点编号，遍历排序后的边，查找这个边上的两个不同的点是否是同一个集合（两个点所属的集合不相同时说明在无向图中将这条边连接不会形成环），若是则继续遍历下一条边否则将这两个点并入同一个集合表示在无向图中将这一条边连接，然后增加这条边的美丽度再继续遍历下一条边

 具体代码：

#include<stdio.h>
#include<stdlib.h>
int n, m, k;
#define size 123456
struct bian { //边结构体
	int u;
	int v;
	int w;
} bb[size];
int cmp(const void*p2, const void *p1) { //给结构体排序的函数
	return ((struct bian *)p1)->w - ((struct bian *)p2)->w;
}
int connect[size];
void init(int n) {
	for (int i = 1; i <= n; i++) {
		connect[i] = i;
	}
}
int find(int n) {
	return connect[n] == n ? n : connect[n] = find(connect[n]);
}
void unions(int a, int b) {
	connect[find(b)] = find(a);
}
int main() {
	scanf("%d %d %d", &n, &m, &k);
	init(n);
	for (int i = 0; i < m; i++) {
		scanf("%d %d %d", &bb[i].u, &bb[i].v, &bb[i].w);
	}
	qsort(bb, m, sizeof(struct bian), cmp);
	int num = 0; //并入最小树的边的数量
	int all = 0; //所有的美丽度
	for (int i = 0; i < m && num < k; i++) {
		if (find(bb[i].u) != find(bb[i].v)) { //这个边可以连入最小树
			unions(bb[i].u, bb[i].v);
			all += bb[i].w;
			num++;
		}
	}
	printf("%d", all);
	return 0;
}