[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
;
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());
}