题目
题解
刷水题一直不好意思写单篇题解,直到我看见了这道题。
把百度搜出来的题解看了前两页。全是二分答案,我才不会告诉你这题,不用二分。当然我一开始也觉得是道二分水题。不过其实不二分也比较水,因为二分能过所以大家没有想太多,二分也是经典套路。
我写到check的时候犹豫了。check在干啥?把cost <=mid 的一级边先加进去,然后加剩下的二级边看能否连通,我怎么觉得这每次做的动作差不多啊。于是诞生了不用二分的想法,若一级边为a边,二级边为b边,a,b先排序,然后先加k条a边进图,注意,这里不能只加前k小的a边,因为有的a边加不进去,要成功加入k次才行。加完之后,两个pointer,分别表示a,b边加到哪里了,在不连通的时候,每次加代价比较小的边,直到连通。其实你有没有发现,先加k条a边进图之后,这就是个最小生成树了,因为限制条件没了,妙。
代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=10000+10;
const int maxm=20000+10;
struct Edge{
int u,v,c;Edge(int u=0,int v=0,int c=0):u(u),v(v),c(c){}
inline bool operator < (const Edge &rhs)const{return c<rhs.c;}
}e1[maxm],e2[maxm];
int p[maxn],cnt;
int findset(int x){return p[x]==x?x:p[x]=findset(p[x]);}
inline void merge(int u,int v)
{
u=findset(u);v=findset(v);
if(u!=v) p[u]=v,--cnt;
}
int main()
{
int n,k,m;cin>>n>>k>>m;cnt=n;--m;
if(n==1) return puts("0"),0;
for(int i=1,u,v,c1,c2;i<=m;++i)
{
scanf("%d%d%d%d",&u,&v,&c1,&c2);
e1[i]=Edge(u,v,c1);e2[i]=Edge(u,v,c2);
}
sort(e1+1,e1+m+1);sort(e2+1,e2+m+1);
for(int i=1;i<=n;++i) p[i]=i;
int p1=1,p2=1;
while(cnt != n-k) merge(e1[p1].u,e1[p1].v),++p1;
while(cnt!=1)
{
if(p2==m+1 || e1[p1].c < e2[p2].c) merge(e1[p1].u,e1[p1].v),++p1;
else merge(e2[p2].u,e2[p2].v),++p2;
}
cout<<max(e1[p1-1].c,e2[p2-1].c)<<endl;
return 0;
}