【bzoj2561】最小生成树

Description

　给定一个边带正权的连通无向图G=(V,E)，其中N=|V|，M=|E|，N个点从1到N依次编号，给定三个正整数u，v，和L (u≠v)，假设现在加入一条边权为L的边(u,v)，那么需要删掉最少多少条边，才能够使得这条边既可能出现在最小生成树上，也可能出现在最大生成树上？

Input

　　第一行包含用空格隔开的两个整数，分别为N和M；
接下来M行，每行包含三个正整数u，v和w表示图G存在一条边权为w的边(u,v)。
最后一行包含用空格隔开的三个整数，分别为u，v，和 L；
数据保证图中没有自环。
Output

　输出一行一个整数表示最少需要删掉的边的数量。

Sample Input

3 2
3 2 1
1 2 3
1 2 2
Sample Output

1
HINT

对于20%的数据满足N ≤ 10，M ≤ 20，L ≤ 20；
对于50%的数据满足N ≤ 300，M ≤ 3000，L ≤ 200；
对于100%的数据满足N ≤ 20000，M ≤ 200000，L ≤ 20000。

题解
加入的边为u,v长度L
所有长度大于L的边不能使得u，v连通
求个最小割

代码

#include<bits/stdc++.h>
#define inf 0x7fffffff
#define N 20005
#define M 400005
using namespace std;
inline 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 ans,n,m,S,T,L,tot=1;
int Head[N],ret[M],Next[M],len[M],cur[N],dis[N];
struct node{int x,y,l;}a[200005];
inline bool cmp(node a,node b){return a.l<b.l;}
inline void ins(int u,int v,int l)
{
    ret[++tot]=v;len[tot]=l;
    Next[tot]=Head[u];Head[u]=tot;
}
inline void insert(int u,int v){ins(u,v,1);ins(v,u,1);}
bool dfs()
{
    for (int i=1;i<=n;i++) dis[i]=-1;queue<int>q;
    q.push(S);dis[S]=0;
    while (!q.empty())
    {
        int now=q.front();q.pop();
        for (int i=Head[now];i;i=Next[i])
            if (dis[ret[i]]==-1&&len[i])
            {
                dis[ret[i]]=dis[now]+1;
                q.push(ret[i]);
            }
    }
    if (dis[T]==-1) return 0;return 1;
}
int dfs(int u,int aug)
{
    if (u==T||aug==0) return aug;
    int flow=0;
    for (int i=cur[u];i;i=Next[i])
        if (dis[ret[i]]==dis[u]+1)
        {
            int t=dfs(ret[i],min(aug-flow,len[i]));
            len[i]-=t;len[i^1]+=t;flow+=t;
            if(len[i])cur[u]=i;
            if (flow==aug) return flow;
        }
    if (flow==0) dis[u]=-1;
    return flow;
}
void dinic()
{
    while (dfs())
    {
        for(int i=1;i<=n;i++)cur[i]=Head[i];
        ans+=dfs(S,inf);
    }
}
int main()
{
    n=read();m=read();
    for (int i=1;i<=m;i++)
    {
        a[i].x=read();a[i].y=read();a[i].l=read();
    }
    S=read();T=read();L=read();
    sort(a+1,a+m+1,cmp);
    for (int i=1;i<=m;i++)if (a[i].l<L) insert(a[i].x,a[i].y);else break;
    dinic();
    for (int i=1;i<=n;i++)Head[i]=0;
    for (int i=m;i;i--) if (a[i].l>L) insert(a[i].x,a[i].y);else break;
    dinic();
    printf("%d",ans);
    return 0;

}