【bzoj2561】最小生成树

嗯……这题是一个网络流。

加入的边为u,v长度L

则所有长度大于L的边不能使得u，v连通

求个最小割即可。小于同理

两次最小割结果相加。

#include<bits/stdc++.h>
#define N 200005
#define M 1000005
#define inf 1000000007
using namespace std;
int tu,tv,tval,n,m,cnt,ans,tot=,head[N],s,t;
struct Edge1{int u,v,w;}T[N];
struct Edge2{int u,v,f,next;}G[M];
inline void addedge(int u,int v,int f){
    G[tot].u=u;G[tot].v=v;G[tot].f=f;G[tot].next=head[u];head[u]=tot++;
    G[tot].u=v;G[tot].v=u;G[tot].f=f;G[tot].next=head[v];head[v]=tot++;
}
inline bool operator<(Edge1 a,Edge1 b){return a.w<b.w;}
int level[N];
bool bfs(int s,int t){
    memset(level,,sizeof(level));queue<int>q;
    q.push(s);level[s]=;
    while(!q.empty()){
        int u=q.front();q.pop();
        if(u==t)return ;
        for(int i=head[u];~i;i=G[i].next){
            int v=G[i].v,f=G[i].f;
            if(f&&!level[v])level[v]=level[u]+,q.push(v);
        }
    }
    return ;
}
int dfs(int u,int maxf,int t){
    if(u==t)return maxf;int rat=;
    for(int i=head[u];~i;i=G[i].next){
        int v=G[i].v,f=G[i].f;
        if(f&&level[v]==level[u]+){
            f=dfs(v,min(maxf-rat,f),t);
            rat+=f;G[i].f-=f;G[i^].f+=f;
        }
    }
    if(!rat)level[u]=inf;
    return rat;
}
inline int dinic(int s,int t){
    int ans=;
    while(bfs(s,t))ans+=dfs(s,inf,t);
    return ans;
}
inline int read(){
    int f=,x=;char ch;
    do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
    do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
    return f*x;
}
int main(){
    memset(head,-,sizeof(head));tot=;
    n=read();m=read();
    for(int i=;i<=m;i++)T[i].u=read(),T[i].v=read(),T[i].w=read();
    tu=read();tv=read();tval=read();
    sort(T+,T+m+);
    for(int i=;i<=m;i++)
    if(T[i].w<tval)addedge(T[i].u,T[i].v,);else break;
    ans+=dinic(tu,tv);
    memset(head,-,sizeof(head));tot=;
    for(int i=m;i;i--)if(T[i].w>tval)addedge(T[i].u,T[i].v,);else break;
    ans+=dinic(tu,tv);
    printf("%d\n",ans);
}