【c++图论】洛谷P2121拆地毯

120 阅读2分钟

学习目标:持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第22天,点击查看活动详情

题目背景

还记得 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<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,k;
int ecnt=0;
struct edge{
	int u,v,w;//记录u(起点),v(终点),w(路线权值)
	bool operator <(const edge &a)const{
		return w>a.w;//以结构体中的w作为排序比较大小的对象
	}
}E[1010101];
void addedge(int u,int v,int w){
	E[++ecnt].u=u;
	E[ecnt].v=v;
	E[ecnt].w=w;
}//记录信息
int f[1010101];
int getf(int x){
	return x==f[x]?x:f[x]=getf(f[x]);找到共同的"爹",(共同的起点)
}
int kruskal(){
	int cnt=0,w=0;//记录当前处理的边数和答案
	sort(E+1,E+1+ecnt);//记得加上<algorithm>头文件
	for(int i=1;i<=ecnt;i++){
		int u=getf(E[i].u);
		int v=getf(E[i].v);
		if(getf(u)!=getf(v)){//如果两个点爹不同,就说明没有相连,这时候需要加边,同时记录答案
			cnt++;
			w+=E[i].w;
			f[u]=v;//同步两个点的爹


		}
		if(cnt==k){//由题意
			break;
		}
	}
	return w;
}
int main(){
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++){
		f[i]=i;//初始化,刚开始每个点的爹都是他自己;
	}
	for(int i=1;i<=m;i++){
		int a,b,c;
		cin>>a>>b>>c;
		addedge(a,b,c);//记录数据
	}
	int c=kruskal();开始处理;
	cout<<c;
}

这道题也是一道模板题,但是需要注意题目中所给的k,在原来模板上稍加改动即可