【bzoj 2561】最小生成树（最小割）

传送门biu~
根据Kruskal算法可知，比最小生成树上的边(u,v,w)权值小的所有边无法将u和v联通。那么在图中加入权值比L小的所有边，跑最小割就是这条边出现在最小生成树上时需要删掉最少的边数；相同的方式计算最大生成树，将两个答案相加就可以了。

#include<bits/stdc++.h>
using namespace std;
const int INF=1e9;
int n,m,S,T,w,ans;
struct Edge{int u,v,w;}e[200005];
int fir[20005],head[20005],dep[20005],nex[400005],to[400005],cap[400005],tp=1;
inline void add(int x,int y,int c){
    nex[++tp]=head[x];
    head[x]=tp;
    to[tp]=y;
    cap[tp]=c;
}
inline int bfs(){
    memset(dep,0,sizeof(dep));
    dep[S]=1;queue<int>q;
    q.push(S);
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=head[x];i;i=nex[i]){
            if(cap[i] && !dep[to[i]]){
                dep[to[i]]=dep[x]+1;
                q.push(to[i]);
            }
        }
    }
    return dep[T];
}
int dfs(int x,int now){
    if(x==T || !now)    return now;
    int c=0;
    for(int &i=fir[x];i;i=nex[i]){
        if(dep[to[i]]==dep[x]+1 && cap[i]){
            int f=dfs(to[i],min(now,cap[i]));
            now-=f;
            cap[i]-=f;
            cap[i^1]+=f;
            c+=f;
            if(!now)    break;
        }
    }
    return c;
}
inline int Dinic(){
    int c=0;
    while(bfs()){
        for(int i=1;i<=n;++i)   fir[i]=head[i];
        c+=dfs(S,INF);
    }
    return c;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;++i)
        scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
    scanf("%d%d%d",&S,&T,&w);
    for(int i=1;i<=m;++i){
        if(e[i].w<w){
            add(e[i].u,e[i].v,1);
            add(e[i].v,e[i].u,1);
        }
    }
    ans+=Dinic();
    memset(head,0,sizeof(head));tp=1;
    for(int i=1;i<=m;++i){
        if(e[i].w>w){
            add(e[i].u,e[i].v,1);
            add(e[i].v,e[i].u,1);
        }
    }
    ans+=Dinic();
    printf("%d",ans);
    return 0;
}