P2323 [HNOI2006] 公路修建问题

题目描述

OI island是一个非常漂亮的岛屿,自开发以来,到这儿来旅游的人很多。然而，由于该岛屿刚刚开发不久，所以那里的交通情况还是很糟糕。所以，OIER Association组织成立了，旨在建立OI island的交通系统。 OI island有n个旅游景点，不妨将它们从1到n标号。现在，OIER Association需要修公路将这些景点连接起来。一条公路连接两个景点。公路有，不妨称它们为一级公路和二级公路。一级公路上的车速快，但是修路的花费要大一些。 OIER Association打算修n-1条公路将这些景点连接起来（使得任意两个景点之间都会有一条路径）。为了保证公路系统的效率, OIER Association希望在这n-1条公路之中,至少有k条(0≤k≤n-1)一级公路。OIER Association也不希望为一条公路花费的钱。所以，他们希望在满足上述条件的情况下，花费最多的一条公路的花费尽可能的少。 而你的任务就是，在给定一些可能修建的公路的情况下，选择n-1条公路，满足上面的条件。

输入描述 Input Description

　　第一行有三个数n,k,m，这些数之间用空格分开。 N和k如前所述，m表示有m对景点之间可以修公路。以下的m行，每一行有4个正整数a,b,c1,c2表示在景点a与b 之间可以修公路,如果修一级公路,则需要c1的花费,如果修二级公路,则需要c2的花费。

输出描述 Output Description

　　一个数据，表示花费最大的公路的花费。

样例输入 Sample Input

4 2 5
1 2 6 5
1 3 3 1
2 3 9 4
2 4 6 1
3 4 4 2

样例输出 Sample Output

4

传送门

题意

这道题的题意首先要读懂。
简单说，就是给你n个点，m-1条边，从中选出n-1条边使任意两点之间都可以到达，需要考虑的是每条边有两种花费，一级公路和二级公路。要求选出k条一级公路，其他任选，使所选边的最大边权值最小。

分析

根据题目易得贪心思路，先按一级公路花费排序，选出n-1条边，并标记，之后剩余的边按二级公路花费排序，加入二级公路的边，并标记。整个过程需要用最小生成树来实现，选边时需要增加对应数组标记公路的级别，直到最小生成树算法结束。

代码

具体代码如下：

#include <bits/stdc++.h>
using namespace std;
const int N=2e4+50;
int n,k,m,ans;
int fa[N],flag[N],cnt;
struct node{
	int a,b,c1,c2,num;
}e[N];
int find(int x){
	if(x==fa[x]) return x;
	return fa[x]=find(fa[x]);
}
bool cmp1(node a,node b){
	return a.c1<b.c1;
}
bool cmp2(node a,node b){
	return a.c2<b.c2;
}
signed main(){
	cin>>n>>k>>m;
	m--;
	for(int i=1;i<=n;i++) 
		fa[i]=i;
	for(int i=1;i<=m;i++){
		cin>>e[i].a>>e[i].b>>e[i].c1>>e[i].c2;
		e[i].num=i;
	}
	sort(e+1,e+m+1,cmp1);
	for(int i=1;i<=m;i++){
		int p=find(e[i].a),q=find(e[i].b);
		if(p!=q){
			fa[p]=q;
			ans=max(ans,e[i].c1);
			cnt++;
			flag[e[i].num]=1;
		}
		if(cnt==k){
			k=i;
			break;
		}
	}
	sort(e+k+1,e+m+1,cmp2);
	for(int i=k+1;i<=m;i++){
		int p=find(e[i].a),q=find(e[i].b);
		if(p!=q){
			fa[p]=q;
			ans=max(ans,e[i].c2);
			flag[e[i].num]=2;
		}
	}
	cout<<ans<<"\n";
	for(int i=1;i<=m;i++)
		if(flag[i]>=1) cout<<i<<" "<<flag[i]<<"\n";
	return 0;
}