2654: tree

xiaoxiao2021-02-28  51

题目链接

题目大意:给你一个无向带权连通图,每条边是黑色或白色。让你求一棵最小权的恰好有need条白色边的生成树。题目保证有解。

题解:容易发现,最小生成树中的白边数目随着白边边权增大而减少。二分一个x,给每条白边都加x,就可以愉快的搞了。为了使二分进行下去,条件是白边数>=need即可。当然,有可能一直二分到最后也不会出现恰好有k条白边的情况,但题目保证有解,这样其实是有很多权相等的黑白边,不用处理

我的收获:强啊

#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; }
转载请注明原文地址: https://www.6miu.com/read-96270.html

最新回复(0)