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*2struct 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;}