sgu176. Flow construction 有源汇上下界最小流

今天一直1A真是爽快

---

Part4：
有源汇上下界最小流。
我们先不连$T->S$的边，然后跑一遍$SS->TT$的最大流，再加上$T->S$跑最大流，答案为边$S->T$的流量。
大概就是一个退流的意思……我并不能表达明白……慢慢理解吧。

#include<iostream>
#include<cstdio>
#include<cstring>
#define inf 1000000007
#define N 105
#define M 20005
using namespace std;
int head[N],du[N],cur[N],dis[N],q[N];
int next[M],list[M],key[M],ans[M];
int n,m,S,T,SS,TT,cnt,tot,total;
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void insert(int x,int y,int z)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
    key[cnt]=z;
}
inline bool BFS()
{
    int t=0,w=1,x;
    memset(dis,-1,sizeof(dis));
    q[1]=SS; dis[SS]=1;
    while (t<w)
    {
        x=q[++t];
        for (int i=head[x];i;i=next[i])
            if (key[i]&&dis[list[i]]==-1)
                dis[q[++w]=list[i]]=dis[x]+1;
    }
    return dis[TT]!=-1;
}
int find(int x,int flow)
{
    if (x==TT) return flow;
    int w,used=0;
    for (int i=cur[x];i;i=next[i])
        if (key[i]&&dis[list[i]]==dis[x]+1)
        {
            w=find(list[i],min(key[i],flow-used));
            key[i]-=w; key[i^1]+=w; used+=w;
            if (key[i]) cur[x]=i;
            if (used==flow) return used;
        }
    if (!used) dis[x]=-1;
    return used;
}
inline int dinic()
{
    int ans=0;
    while (BFS())
    {
        for (int i=0;i<=TT;i++) cur[i]=head[i];
        ans+=find(SS,inf);
    }
    return ans;
}
int main()
{
    n=read(); m=read(); S=1; T=n; SS=0; TT=n+1; cnt=1;
    for (int i=1;i<=m;i++)
    {
        int u=read(),v=read(),w=read(),c=read();
        if (!c) insert(u,v,w),insert(v,u,0);
        else du[u]-=w,du[v]+=w,ans[i]=w;
    }
    for (int i=1;i<=n;i++) 
        if (du[i]>0) total+=du[i],insert(SS,i,du[i]),insert(i,SS,0);
        else if (du[i]<0) insert(i,TT,-du[i]),insert(TT,i,0);
    tot+=dinic(); insert(T,S,inf); insert(S,T,0); tot+=dinic();
    if (tot!=total) puts("Impossible"); 
    else 
    {
        printf("%d\n",key[cnt]);
        int now=0;
        for (int i=1;i<=m;i++)
            if (!ans[i]) ans[i]=key[(++now)<<1|1];
        for (int i=1;i<=m;i++) printf("%d ",ans[i]);
    }
    return 0;
}