#116. 有源匯有上下界最大流

 

題目描述

這是一道模板題。

n個點，m  條邊，每條邊 e 有一個流量下界 lower(e)  和流量上界 upper(e) )，給定源點 s 與匯點 t，求源點到匯點的最大流。

輸入格式

第一行兩個正整數 n、m、s、t。

以後的 m 行，每行四個整數 s、t、lower、upper。

輸出格式

若是無解，輸出一行 please go home to sleep。

不然輸出最大流。

樣例

樣例輸入

10 15 9 10
9 1 17 18
9 2 12 13
9 3 11 12
1 5 3 4
1 6 6 7
1 7 7 8
2 5 9 10
2 6 2 3
2 7 0 1
3 5 3 4
3 6 1 2
3 7 6 7
5 10 16 17
6 10 10 11
7 10 14 15

樣例輸出

43

數據範圍與提示

1≤n≤202,1≤m≤9999

題解：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#define inf 0x3f3f3f3f
#define ll long long
#define MAXN 30000
using namespace std;
int n,m;//點數、邊數
int X[MAXN],y[MAXN];
int sp,tp;//原點、匯點
struct node
{
    int v,next;
    ll cap;
}mp[MAXN*10];
int pre[MAXN],dis[MAXN],cur[MAXN];//cur爲當前弧優化，dis存儲分層圖中每一個點的層數（即到原點的最短距離），pre建鄰接表
int cnt=0;
void init()//不要忘記初始化
{
    cnt=0;
    memset(pre,-1,sizeof(pre));
}
void add(int u,int v,int w)//加邊
{
    mp[cnt].v=v;
    mp[cnt].cap=w;
    mp[cnt].next=pre[u];
    pre[u]=cnt++;
    mp[cnt].v=u;
    mp[cnt].cap=0;
    mp[cnt].next=pre[v];
    pre[v]=cnt++;
}
bool bfs()//建分層圖
{
    memset(dis,-1,sizeof(dis));
    queue<int>q;
    while(!q.empty())
        q.pop();
    q.push(sp);
    dis[sp]=0;
    int u,v;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(int i=pre[u];i!=-1;i=mp[i].next)
        {
            v=mp[i].v;
            if(dis[v]==-1&&mp[i].cap>0)
            {
                dis[v]=dis[u]+1;
                q.push(v);
                if(v==tp)
                    break;
            }
        }
    }
    return dis[tp]!=-1;
}
ll dfs(int u,ll cap)//尋找增廣路
{
    if(u==tp||cap==0)
        return cap;
    ll res=0,f;
    for(int &i=cur[u];i!=-1;i=mp[i].next)
    {
        int v=mp[i].v;
        if(dis[v]==dis[u]+1&&(f=dfs(v,min(cap-res,mp[i].cap)))>0)
        {
            mp[i].cap-=f;
            mp[i^1].cap+=f;
            res+=f;
            if(res==cap)
                return cap;
        }
    }
    if(!res)
        dis[u]=-1;
    return res;
}
ll dinic()
{
    ll ans=0;
    while(bfs())
    {
        for(int i=0;i<=tp;i++)
            cur[i]=pre[i];
        ans+=dfs(sp,inf);
    }
    return ans;
}
int d[MAXN];
int main()
{
    init();
    int s,t;
    scanf("%d%d%d%d",&n,&m,&s,&t);
    int x,y,low,up;
    for (int i = 0; i <m ; ++i) {
        scanf("%d%d%d%d",&x,&y,&low,&up);
        add(x,y,up-low);
        d[x]-=low;
        d[y]+=low;
    }
    int sum=0;
    for (int i = 1; i <=n ; ++i) {
        if(d[i]>0)
        {
            sum+=d[i];
            add(0,i,d[i]);
        }
        if(d[i]<0)
            add(i,n+1,-d[i]);
    }
    add(t,s,inf);
    sp=0;tp=n+1;
    int k=dinic();
    if(k!=sum)
    {
        printf("please go home to sleep\n");
    } else
    {
        sp=s;tp=t;
        printf("%lld\n",dinic());
    }
    return 0;
}