BZOJ1196: [HNOI2006]公路修建问题

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAXN 20010
using namespace std;
int n,m,k,ans=0;
int fa[MAXN];
struct Edge{
	int u,v,w1,w2;
}a[MAXN];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<‘0‘||c>‘9‘){if(c==‘-‘)w=-1;c=getchar();}
	while(c>=‘0‘&&c<=‘9‘){date=date*10+c-‘0‘;c=getchar();}
	return date*w;
}
inline bool cmp1(const Edge &p,const Edge &q){
	if(p.w1==q.w1)return p.w2>q.w2;
	return p.w1<q.w1;
}
inline bool cmp2(const Edge &p,const Edge &q){
	return p.w2<q.w2;
}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
inline void uniun(int x,int y){x=find(x);y=find(y);if(x!=y)fa[y]=x;}
void kruskal(){
	int s=0,now=1;
	sort(a+1,a+m+1,cmp1);
	for(int i=1;i<=m&&s<k;i++)
	if(find(a[i].u)!=find(a[i].v)){
		uniun(a[i].u,a[i].v);
		ans=max(ans,a[i].w1);
		s++;
		now++;
	}
	sort(a+now,a+m+1,cmp2);
	for(int i=now;i<=m&&s<n-1;i++)
	if(find(a[i].u)!=find(a[i].v)){
		uniun(a[i].u,a[i].v);
		ans=max(ans,a[i].w2);
		s++;
	}
	printf("%d\n",ans);
}
void init(){
	n=read();k=read();m=read()-1;
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=m;i++){a[i].u=read();a[i].v=read();a[i].w1=read();a[i].w2=read();}
}
int main(){
	init();
	kruskal();
    return 0;
}