BZOJ-2654: tree (kruskal)

2654: tree

Time Limit: 30 Sec  Memory Limit: 512 MB
Submit: 2398  Solved: 1003
[Submit][Status][Discuss]

Description

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

题目保证有解。

Input

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

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

Output

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

V<=50000,E<=100000,所有数据边权为[1,100]中的正整数。

Sample Input

2 2 1
0 1 1 1
0 1 2 0

Sample Output

2

HINT

原数据出错，现已更新 by liutian,但未重测---2016.6.24

Source

一开始竟然忘记给边排序了……还有在排序的过程中权值相同的白色的边排在前面！！！就是因为这个地方WA了一次QAQ

#include "bits/stdc++.h"
using namespace std;
typedef long long LL;
const int MAX1=5e5+5;
const int MAX2=1e6+5;
int n,m,need,tmp,ans;
int fa[MAX1];
struct Node{
    int u,v;
    int col,wei;
    bool operator < (const Node &tt) const {
        if (wei!=tt.wei) return wei<tt.wei;
        else return col<tt.col;//非常重要！！！ 
    }
}a[MAX2],b[MAX2];
inline int read(){
    int an=0,x=1;char c=getchar();
    while (c<'0' || c>'9') {if (c=='-') x=-1;c=getchar();}
    while (c>='0' && c<='9') {an=an*10+c-'0';c=getchar();}
    return an*x;
}
int getfather(int x){ return fa[x]==x?x:fa[x]=getfather(fa[x]);}
bool feasible(int x){
    int i,j,zt;
    tmp=zt=0;
    for (i=1;i<=n;i++) fa[i]=i;
    for (i=1;i<=m;i++){
        b[i]=a[i];
        if (!b[i].col) b[i].wei+=x;
    }
    sort(b+1,b+m+1);
    for (i=1;i<=m;i++){
        int tx=getfather(b[i].u);
        int ty=getfather(b[i].v);
        if (tx!=ty){
            fa[tx]=ty;
            tmp+=b[i].wei;
            if (!b[i].col)zt++;
        }
    }
    return zt>=need;
}
int main(){
    freopen ("tree.in","r",stdin);freopen ("tree.out","w",stdout);
    int i,j,u,v,col,wei;
    n=read();m=read();need=read();
    for (i=1;i<=m;i++) a[i].u=read(),a[i].v=read(),a[i].wei=read(),a[i].col=read(),a[i].u++,a[i].v++;
    int low=-100,high=100,mid;
    while (low<=high){
        mid=(low+high)>>1;
        if (feasible(mid)) ans=tmp-need*mid,low=mid+1;
        else high=mid-1;
    }
    printf("%d",ans);
    return 0;
}