【洛谷P2323】【gmoj1255】【8.11模拟赛T2】公路修建问题【kruskal+贪心】

题目链接：gmoj、洛谷

分析

这题一眼看就像是最小生成树，但又好像不完全是。。一开始想着裸的贪心好像也不太可行。

于是，美妙的结合来了。

因为对于一级公路来说最大的一级公路一定是全部公路中最大的。而如果一个二级公路很大，所对应一级公路一定更大。所以就符合贪心啦。
只要选取最小的k条一级公路就可以。如果换成任意一个一级公路都会更大。
而且要记录这k条一级公路连通了几个点，这里用kruskal实现。
然后剩下没有连通的点，再排序二级公路的大小，用最小生成树实现连通。这里并不会选到重复的边，因为选过记录了连通就不会再选。

上代码

gmoj

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

struct node
{
	int u,v,c1,c2;
}a[50001];

int n,k,m,fa[100001],ans1,ans2,ans,ff=0;

int cmp1(node l,node r)
{
    return l.c1<r.c1;
}

int cmp(node l,node r)
{
	return l.c2<r.c2;
}

int father(int x)
{
	if(fa[x]==x) return x;
	else return father(fa[x]);
}

int main()
{
	cin>>n>>k>>m;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d%d",&a[i].u,&a[i].v,&a[i].c1,&a[i].c2);	
	}
	sort(a+1,a+m+1,cmp1);
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		int fx=father(a[i].u);
		int fy=father(a[i].v); 
		if(fx!=fy)
		{
		    fa[fx]=fy;
			ans1=max(ans1,a[i].c1);	
			ff++;
		}
		if(ff==k) break;
	}
	sort(a+1,a+m+1,cmp);
	for(int i=1;i<=m;i++)
	{
		int fx=father(a[i].u);
		int fy=father(a[i].v); 
		if(fx!=fy)
		{
		    fa[fx]=fy;
			ans2=max(ans2,a[i].c2);	
			ff++;
		}
		if(ff==n-1) break;
	}
	ans=max(ans1,ans2);
	cout<<ans;
	return 0;
} 

洛谷

加两个记录就可以。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

struct node
{
	int u,v,c1,c2,id;
}a[50001];

struct lwx
{
	int t,p;
}b[100001];

int n,k,m,fa[100001],ans1,ans2,ans,ff=0;

int cmp1(node l,node r)
{
    return l.c1<r.c1;
}

int cmp(node l,node r)
{
	return l.c2<r.c2;
}

int cmpp(lwx l,lwx r)
{
	return l.t<r.t;
}

int father(int x)
{
	if(fa[x]==x) return x;
	else return father(fa[x]);
}

int main()
{
	cin>>n>>k>>m;
	for(int i=1;i<m;i++)
	{
		scanf("%d%d%d%d",&a[i].u,&a[i].v,&a[i].c1,&a[i].c2);	
		a[i].id=i;
	}
	sort(a+1,a+m+1,cmp1);
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		int fx=father(a[i].u);
		int fy=father(a[i].v); 
		if(fx!=fy)
		{
		    fa[fx]=fy;
		    b[++ff].p=1;
		    b[ff].t=a[i].id;
			ans1=max(ans1,a[i].c1);	
		}
		if(ff==k) break;
	}
	sort(a+1,a+m+1,cmp);
	for(int i=1;i<=m;i++)
	{
		int fx=father(a[i].u);
		int fy=father(a[i].v); 
		if(fx!=fy)
		{
		    fa[fx]=fy;
		    b[++ff].p=2;
		    b[ff].t=a[i].id;
			ans2=max(ans2,a[i].c2);	
		}
		if(ff==n-1) break;
	}
	ans=max(ans1,ans2);
	cout<<ans<<endl;
	sort(b+1,b+n-1+1,cmpp);
	for(int i=1;i<=n-1;i++)
	{
		cout<<b[i].t<<' '<<b[i].p<<endl;
	} 
	return 0;
}