1196: [HNOI2006]公路修建问题

题目链接

题目大意：修n-1条边将n个点连通，每个点可建一级边也可建二级边(一级边费用高)，要求一级边必须有k条，使花费最多的公路花费最少。

题解：答案具有单调性，考虑二分答案。具体检验过程：1.把费用小于mid且可以放进生成树的1级边选上 2.如果步骤1选出足够的1级边，判断能否构成生成树

我的收获：条件的转化

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <climits> 
using namespace std;
#define INF INT_MAX/2
const int M=10005;
int n,m,l,r,mid,ans,k,x,y;
int w[M],f[M];

struct edge{int u,v,c1,c2;}e[M*5];

int find(int x){return f[x]==x?x:f[x]=find(f[x]);}

bool ok(int mx)
{
    int cnt=0;//1级公路条数
    for(int i=1;i<=n;i++) f[i]=i;//初始化
    for(int i=1;i<=m;i++)
    {
        if(e[i].c1>mx) continue;
        x=find(e[i].u),y=find(e[i].v);
        if(x!=y){cnt++;f[x]=y;}
    }
    if(cnt<k) return false;
    for(int i=1;i<=m;i++)
    {
        if(e[i].c2>mx) continue;
        x=find(e[i].u),y=find(e[i].v);
        if(x!=y){cnt++;f[x]=y;}
    }
    if(cnt==n-1) return true;
    return false;
}

void work()
{
    l=1,r=100000;
    while(l<=r){
        mid=(l+r)>>1;
        if(ok(mid))
        r=mid-1,ans=mid;
        else l=mid+1;
    }
    printf("%d\n",ans);
} 

void init()
{
    cin>>n>>k>>m;
    for(int i=1;i<=m;i++)
        scanf("%d%d%d%d",&e[i].u,&e[i].v,&e[i].c1,&e[i].c2);
}

int main()
{
    init();
    work();
    return 0;
}