BZOJ1196: [HNOI2006]公路修建问题

题目

BZOJ1196: [HNOI2006]公路修建问题

题解

刷水题一直不好意思写单篇题解，直到我看见了这道题。
把百度搜出来的题解看了前两页。全是二分答案，我才不会告诉你这题，不用二分。当然我一开始也觉得是道二分水题。不过其实不二分也比较水，因为二分能过所以大家没有想太多，二分也是经典套路。

我写到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;
}