码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ 2654 tree

时间:2016-09-06 12:09:59      阅读:119      评论:0      收藏:0      [点我收藏+]

标签:

二分+kruskal。

为什么老是WA?

23333不管了粘黄学长代码。

#include<iostream>
#include<set>
#include<map>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
#define inf 1000000000
#define pa pair<int,int>
#define ll long long 
using namespace std;
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 n,m,cnt,tot,ned;
int sumv;
int u[100005],v[100005],w[100005],c[100005];
int fa[100005];
struct edge{
    int u,v,w,c;
}e[100005];
bool operator<(edge a,edge b)
{
    return a.w==b.w?a.c<b.c:a.w<b.w;
}
int find(int x)
{
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
bool check(int x)
{
    tot=cnt=0;
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        e[i].u=u[i],e[i].v=v[i],e[i].w=w[i];e[i].c=c[i];
        if(!c[i])e[i].w+=x;
    }
    sort(e+1,e+m+1);
    for(int i=1;i<=m;i++)
    {
        int p=find(e[i].u),q=find(e[i].v);
        if(p!=q)
        {
            fa[p]=q;
            tot+=e[i].w;
            if(!e[i].c)cnt++;
        }
    }
    return cnt>=ned;
}
int main()
{
    n=read();m=read();ned=read();
    for(int i=1;i<=m;i++)
    {
        u[i]=read(),v[i]=read(),w[i]=read(),c[i]=read();
        u[i]++;v[i]++;
    }
    int l=-105,r=105;
    while(l<=r)
    {
        int mid=(l+r)>>1;
        if(check(mid))l=mid+1,sumv=tot-ned*mid;
        else r=mid-1;
    }
    printf("%d",sumv);
    return 0;
}

 

BZOJ 2654 tree

标签:

原文地址:http://www.cnblogs.com/ziliuziliu/p/5845039.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!