LOJ 117 有源汇有上下界最小流(模板)

建图方法
将有上下界的网络流图转化成普通的网络流图
首先建立附加源点ss和附加汇点tt
对于原图中的边x->y,若限制为[b,c],那么连边x->y,流量为c-b
对于原图中的某一个点i,记d(i)为流入这个点的所有边的下界和减去流出这个点的所有边的下界和
若d(i)>0,那么连边ss->i,流量为d(i)
若d(i)<0,那么连边i->tt,流量为-d(i)
(以上与无源汇可行流相同)
求res=ss->tt最大流
连边t->s,inf
求res+=ss->tt最大流
如果res!=sum也就是期望中的满流,那么就没有可行解。
需要格外注意最小流判断是否有可行解的位置与时机与另外几种上下界网络流的不同!!!
否则,输出G[s].back().cap也就是从t到s inf边的实际流量,即为所求。
注意本题中数据非常大,必须开longlong不然会爆;

#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long ll;
struct edge{
    int to;ll cap;int rev;
};
vectorG[50210];
int level[50205],iter[50205],low[125050];ll cur[50205];
void addedge(int from,int to,ll cap)
{
    G[from].push_back(edge{to,cap,(int)G[to].size()});
    G[to].push_back(edge{from,0,(int)G[from].size()-1});//反向容量为0!!
}
void bfs(int s)
{
    memset(level,-1,sizeof(level));
    queue<int>que;
    level[s]=0;que.push(s);
    while(!que.empty()){
        int t=que.front();que.pop();
        for(int i=0;iif(e.cap>0&&level[e.to]<0){
                level[e.to]=level[t]+1;
                que.push(e.to);
            }
        }
    }
}
ll dfs(int v,int t,ll f)
{
    if(v==t)return f;
    for(int&i=iter[v];i//注意传引用!
        edge&e=G[v][i];
        if(e.cap>0&&level[v]if(d>0){
                e.cap-=d;
                G[e.to][e.rev].cap+=d;
                return d;
            }
        }
    }
    return 0;//不要漏了这个,很多时候可能是无法增广的
}
ll maxflow(int s,int t){
    ll flow=0;
    for(;;){
        bfs(s);
        if(level[t]<0)return flow;
        memset(iter,0,sizeof(iter));
        ll f;
        while(f=dfs(s,t,0x7f7f7f7f7f7f7f7f))
            flow+=f;
    }
}
int main()
{
    int n,m,s,t,i,j,k;
    cin>>n>>m>>s>>t;
    for(i=1;i<=m;i++){
        int from,to,up;scanf("%d%d%d%d",&from,&to,&low[i],&up);
        cur[from]-=low[i];cur[to]+=low[i];addedge(from,to,up-low[i]);
    }
    ll sum=0;
    for(i=1;i<=n;i++){
        if(cur[i]>0){
            addedge(0,i,cur[i]);sum+=cur[i];
        }
        else if(cur[i]<0){
            addedge(i,50050,-cur[i]);
        }
    }
    ll flow=maxflow(0,50050);
    addedge(t,s,(1LL<<63)-1);//注意加边位置与时刻
    flow+=maxflow(0,50050);
    if(flow!=sum){
        cout<<"please go home to sleep"<return 0;
    }
    cout<return 0;
}

你可能感兴趣的:(模板,图论,网络流)