Luogu2619[国家集训队2] Tree I

原题链接：https://www.luogu.org/problemnew/show/P2619

Tree I

题目描述

给你一个无向带权连通图，每条边是黑色或白色。让你求一棵最小权的恰好有need条白色边的生成树。

题目保证有解。

输入输出格式

输入格式：

第一行V,E,need分别表示点数，边数和需要的白色边数。

接下来E行

每行s,t,c,col表示这边的端点(点从0开始标号)，边权，颜色(0白色1黑色)。

输出格式：

一行表示所求生成树的边权和。

输入输出样例

输入样例#1：

2 2 1
0 1 1 1
0 1 2 0

输出样例#1：

2

说明

0:V<=10

1,2,3:V<=15

0,…,19:V<=50000,E<=100000

所有数据边权为[1,100]中的正整数。

By WJMZBMR

题解

带权二分第一题。。。

如果白色边的权值都加上$+\infty$，最小生成树中的白色边会尽量少；如果白色边权值都加上$-\infty$，最小生成树中的白色边就会尽量多。

所以我们二分白色边的附加权值，每次做$Kruscal$，看白边有几条，最后输出答案即可。

代码

#include
#define inf 100
using namespace std;
const int M=1e5+5;
struct sd{int a,b,val,col;}ed[M],now[M];
bool operator<(sd a,sd b){return a.val==b.val?a.col<b.col:a.val<b.val;}
int f[M],n,m,need,cot,tot,ans;
int root(int v){return f[v]==v?v:f[v]=root(f[v]);}
bool check(int d)
{
    ans=cot=tot=0;
    for(int i=1;i<=n;++i)f[i]=i;
    for(int i=1;i<=m;++i)now[i]=ed[i],now[i].val+=(ed[i].col?0:d);
    sort(now+1,now+1+m);
    for(int i=1;i<=m&&tot<n;++i)if(root(now[i].a)!=root(now[i].b))
    f[f[now[i].a]]=f[now[i].b],cot+=(now[i].col^1),ans+=now[i].val,++tot;
    return cot>=need;
}
void in()
{
    scanf("%d%d%d",&n,&m,&need);
    for(int i=1,a,b,c,d;i<=m;++i)
    scanf("%d%d%d%d",&a,&b,&c,&d),ed[i]=(sd){a+1,b+1,c,d};
}
void ac()
{
    int l=-inf,r=inf,mid,d;
    for(;l<=r;mid=l+r>>1,check(mid)?(l=mid+1,d=mid):r=mid-1);
    check(d);printf("%d",ans-need*d);
}
int main(){in(),ac();}