BZOJ_P2561 最小生成树(网络流+最大流ISAP)

BZOJ传送门
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 1211 Solved: 585
[Submit][Status][Discuss]
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。

Source
2012国家集训队Round 1 day1

这题阴吹思婷啊是不是，小于l的肯定不能再最小生成树里，大于l的肯定不在最大生成树里，分两次搞掉这两个限制就可以了
数据辣么大谁能想到网络流qwq~~

#include<cstdio>
#include<climits>
#include<cstring>
#include<queue>
#include<vector>
#include<iostream>
using namespace std;
#define N 20005
#define M 200005
#define INF INT_MAX/3*2
struct NetWork{
    struct Edge{int fr,to,cap,flow;};
    vector<Edge> edge;vector<int> g[N];
    int cur[N],p[N],d[N],num[N];bool v[N];
    int s,t,n,m,l,ans;

    int u[M],r[M],w[M];

    inline void Add_Edge(int fr,int to,int cap,int flow){
        edge.push_back(Edge{fr,to,cap,flow});
        edge.push_back(Edge{to,fr,cap,flow});int tmp=edge.size();
        g[fr].push_back(tmp-2);g[to].push_back(tmp-1);
    }

    inline int BFS(){
        for(int i=1;i<=n;i++) d[i]=n;
        memset(v,0,sizeof(v));queue<int> q;
        q.push(t);v[t]=true,d[t]=0;int x;
        while(!q.empty()){
            x=q.front();q.pop();
            for(int i=0;i<g[x].size();i++){
                Edge &e=edge[g[x][i]];
                Edge &u=edge[g[x][i]^1];
                if(!v[e.to]&&u.cap>u.flow){
                    d[e.to]=d[x]+1;v[e.to]=true;
                    q.push(e.to);
                }
            }
        }
        return d[s];
    }
    inline int AddFlow(){
        int x=t,a=INF;Edge e=edge[p[x]];
        for(;x!=s;x=e.fr,e=edge[p[x]]) a=min(a,e.cap-e.flow);
        for(x=t;x!=s;x=edge[p[x]].fr) edge[p[x]].flow+=a,edge[p[x]^1].flow-=a;
        return a;
    }
    int MaxFlow(){
        int x=s,flow=0;BFS();
        memset(cur,0,sizeof(cur));memset(num,0,sizeof(num));
        for(int i=1;i<=n;i++) num[d[i]]++;
        while(d[s]<n){
            if(x==t) flow+=AddFlow(),x=s;
            int ok=0;
            for(int i=cur[x];i<g[x].size();i++){
                Edge &e=edge[g[x][i]];
                if(e.cap>e.flow&&d[x]==d[e.to]+1){
                    ok=1,p[e.to]=g[x][i];
                    cur[x]=i,x=e.to;break;
                }
            }
            if(!ok){
                int m=n-1;
                for(int i=0;i<g[x].size();i++){
                    Edge &e=edge[g[x][i]];
                    if(e.cap>e.flow) m=min(m,d[e.to]);
                }
                if(!(--num[d[x]])) break;
                num[d[x]=m+1]++;cur[x]=0;
                if(x!=s) x=edge[p[x]].fr;
            }
        }
        return flow;
    }

    inline void clear(){for(int i=1;i<=n;i++) g[i].clear();edge.clear();}
    inline int in(){
        int x=0;char ch=getchar();
        while(ch>'9'||ch<'0') ch=getchar();
        while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
        return x;
    }
    int solve(){
        n=in();m=in();
        for(int i=1;i<=m;i++) u[i]=in(),r[i]=in(),w[i]=in();
        s=in(),t=in(),l=in();
        for(int i=1;i<=m;i++)
            if(w[i]<l) Add_Edge(u[i],r[i],1,0);
        ans+=MaxFlow();
        clear();
        for(int i=1;i<=m;i++)
            if(w[i]>l) Add_Edge(u[i],r[i],1,0);
        ans+=MaxFlow();
        printf("%d",ans);
    }
}s;
int main(){
    s.solve();
    return 0;
}