[BZOJ1196]HNOI2006公路修建问题|Kruskal|贪心|二分答案

二分答案很显然，贪心我没想到怎么贪。。

    其实也挺显然的。。设当前二分到的答案是ans，将边排序（一级公路第一关键字，二级第二）后类似kruskal那样贪。。从小到大，如果一级公路费用<=ans，那直接选进来，else if二级公路费用<=ans且已选满k条一级公路，那就加进来。。最后如果能选到n-1条边就说明可行。。

#include<cstdio>
#include<iostream>
#include<algorithm>
#define N 10005
using namespace std;
struct edge{
	int s,e,q1,q2;
}ed[N*2];
int n,m,k,i,mq,fa[N];
int getfa(int x){return fa[x]==x?x:fa[x]=getfa(fa[x]);}
bool cmp(edge a,edge b)
{
	if (a.q1!=b.q1) return a.q1<b.q1;else return a.q2<b.q2;
}
bool check(int t)
{
	int cnt=0,tot=0,f1,f2;
	for (i=1;i<=m&&cnt<n-1;i++)
	{
		if (ed[i].q1>t&&tot<k) return false;
		if (ed[i].q1<=t||(ed[i].q2<=t&&tot>=k))
		{
			f1=getfa(ed[i].s);f2=getfa(ed[i].e);
			if (f1!=f2)
			{
				cnt++;
				if (ed[i].q1<=t) tot++;
				fa[f1]=f2;
			}
		}
	}
	if (cnt==n-1&&tot>=k) return true;else return false;
}
int binary()
{
	int l=1,r=mq,mid;
	while (l<r)
	{
		mid=(l+r)>>1;
		for (i=1;i<=n;i++) fa[i]=i;
		if (check(mid)) r=mid;else l=mid+1;
	}
	return l;
}
int main()
{
	scanf("%d%d%d",&n,&k,&m);m--;
	mq=0;
	for (i=1;i<=m;i++)	scanf("%d%d%d%d",&ed[i].s,&ed[i].e,&ed[i].q1,&ed[i].q2),mq=max(mq,ed[i].q1);
	sort(ed+1,ed+m+1,cmp);
	printf("%d\n",binary());
}