BZOJ2654 & 洛谷2619：tree——题解

#include<queue>
#include<cmath>
#include<stack>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=50005;
const int M=100005;
inline int read(){
    int X=0,w=0;char ch=0;
    while(!isdigit(ch)){w|=ch==‘-‘;ch=getchar();}
    while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
struct node{
    int u,v,w,c;
}e[M];
int n,m,t,tmp,fa[N];
int find(int x){
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
}
inline void unionn(int u,int v){
    fa[u]=v;
}
inline bool cmp(node a,node b){
    return a.w<b.w||(a.w==b.w&&a.c<b.c);
}
void modify(int k){
    for(int i=1;i<=m;i++){
        if(!e[i].c)e[i].w+=k;
    }
}
bool pan(int k){
    modify(k);
    sort(e+1,e+m+1,cmp);
    int num=0,cnt=0;tmp=0;
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=m;i++){
        int u=e[i].u,v=e[i].v,w=e[i].w;
        u=find(u),v=find(v);
        if(u!=v){
            unionn(u,v);
            cnt++;tmp+=w;
            if(!e[i].c)num++;
            if(cnt==n-1)break;
        }
    }
    modify(-k);
    return num>=t;
}
int solve(int l,int r){
    int ans;
    while(l<r){
        int mid=(l+r+1)>>1;
        if(!pan(mid))r=mid-1;
        else{
            l=mid;
            ans=tmp-mid*t;
        }
    }
    return ans;
}
int main(){
    n=read(),m=read(),t=read();
    for(int i=1;i<=m;i++){
        e[i].u=read()+1,e[i].v=read()+1;
        e[i].w=read(),e[i].c=read();
    }
    printf("%d\n",solve(-100,100));
    return 0;
}