Tree（指定最小生成树）

原题链接

解题思路：不会^ v ^

C++代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;

const int M=100005;

int n,m,k,tot,cnt;
int f[M];

struct edge{int u,v,w,c;}e[M*2];

int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

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

bool cmp(edge a,edge b){return a.w<b.w||(a.w==b.w&&a.c<b.c);}

bool check(int x)
{
    tot=cnt=0;
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=m;i++) if(!e[i].c) e[i].w+=x;
    sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m;i++)
    {
        int p=fd(e[i].u),q=fd(e[i].v);
        if(p!=q){
            f[p]=q;tot+=e[i].w;
            if(!e[i].c) cnt++;
        }
    }
    for(int i=1;i<=m;i++) if(!e[i].c) e[i].w-=x;
    return cnt>=k;
}

void init()
{
    n=read();m=read();k=read();
    for(int i=1;i<=m;i++)
    e[i].u=read(),e[i].v=read(),e[i].w=read(),e[i].c=read(),e[i].u++,e[i].v++;
}

void work()
{
    int sumv,l=-105,r=105;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(check(mid)) l=mid+1,sumv=tot-k*mid;
        else r=mid-1;
    }
    printf("%d",sumv);
}

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

参考博客