P2121 拆地毯

# 拆地毯

## 题目背景

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

## 题目描述

## 输入格式

第一行包含三个正整数 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
```

## 提示

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

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

1<=n,m,k<=100000

1.这是最大生成树问题，建议使用kruskal。

 2.题目说保留  k  块地毯，于是在地毯数为 k 时最小生成树就构建好了。我们只需要排序（降序），再去看 u (起点) 和 v（终点）是否是处在一个集合当中，如果是，那么这条地毯就不需要，否则就加上它的美丽值。看是否在一个集合中，使用并查集的思想即可——（可以看这个）2023/1/2总结_lxh0113的博客-CSDN博客

 3，具体kruskal算法可以看这个https://blog.csdn.net/lxh0113/article/details/128553299?spm=1001.2014.3001.5502

C代码如下：

#include<stdio.h>
#define N 100000
struct node
{
	int u,v,w;
}edges[N+10];
int b[N+10],n,m,k;
int getf(int x)
{
	if(b[x]==x) return x;
	b[x]=getf(b[x]);
}
int pd(int x,int y)
{
	int p,q;
	p=getf(x);
	q=getf(y);
	if(p!=q)
	{
		b[q]=p;
		return 1;
	}
	return 0;
}
int quicksort(int left,int right)
{
	if(left>=right) return 0;
	int i=left,j=right;
	struct node t,temp=edges[left];
	while(i<j)
	{
		while(i<j&&temp.w>=edges[j].w) j--;
		while(i<j&&temp.w<=edges[i].w) i++;
		if(i<j)
		{
			t=edges[i];edges[i]=edges[j];edges[j]=t;
		}
	}
	edges[left]=edges[i];
	edges[i]=temp;
	quicksort(left,i-1);
	quicksort(i+1,right);
}
 
int kruskal()
{
	int i,j,sum=0;
	for(i=0,j=0;i<m;i++)
	{
		if(pd(edges[i].u,edges[i].v))
		{
			j++;
			sum+=edges[i].w;
		}
		if(j>=k) break;
	}
	return sum;
}
 
int main()
{
	int i,j;
	scanf("%d%d%d",&n,&m,&k);
	for(i=1;i<=n;i++)
		b[i]=i;
	for(i=0;i<m;i++)
	{
		scanf("%d%d%d",&edges[i].u,&edges[i].v,&edges[i].w);
	}
	quicksort(0,m-1);
	printf("%d\n",kruskal());
	return 0;	
}

 C++代码如下：

#include<algorithm>
#include<iostream>
#include<bits/stdc++.h>
 
using namespace std;
 
const int  N = 100000;
struct node 
{
	int u, v, w;
} edges[N+10];
 
int b[N+10], n, m;
int getf(int x) 
{
	if (b[x] == x) return x;
	b[x] = getf(b[x]);
}
int pd(int x, int y) 
{
	int p, q;
	p = getf(b[x]);
	q = getf(b[y]);
	if (p != q) 
	{
		b[q] = p;
		return 1;
	} else return 0;
}
bool cmp(node a, node b) 
{
	return a.w < b.w;
}
int kruskal() 
{
	int i, j, sum = 0;
	for (i = 0, j = 0; i < m; i++) 
	{
		if (pd(edges[i].u, edges[i].v) == 1) 
		{
			j++;
			sum += edges[i].w;
		}
		if (j >= n) break;
	}
	return sum;
}
 
int main() 
{
	int i, j, k;
	cin >> n >> m;
	for (i = 1; i <= n; i++)
		b[i] = i;
	for (i = 0; i < m; i++) 
	{
		cin >> edges[i].u >> edges[i].v >> edges[i].w ;
	}
	sort(edges, edges + m - 1, cmp);
	k = kruskal();
	cout << k << endl;
	return 0;
}