上下界网络流模型常见解法

#include <iostream> 
#include <queue>
#include <cstdio>
#include <cstring>
using namespace std ;

const int INF=0xfffffff ;
struct node
{
    int s,t,cap,nxt ;
}e[400005] ;
int m,n,cnt,head[100005],level[100005],q[100005] ;
void add(int s,int t,int cap)
{
    e[cnt].s=s ;e[cnt].t=t ;e[cnt].cap=cap ;e[cnt].nxt=head[s] ;head[s]=cnt++ ;
    e[cnt].s=t ;e[cnt].t=s ;e[cnt].cap=0 ;e[cnt].nxt=head[t] ;head[t]=cnt++ ;
}
bool build(int s,int t)
{
    int front=0,rear=0 ;
    memset(level,-1,sizeof(level)) ;
    q[rear++]=s ;
    level[s]=1 ;
    while(front<rear)
    {
        int u=q[front++] ;
        for(int i=head[u] ;i!=-1 ;i=e[i].nxt)
        {
            int tt=e[i].t ;
            if(level[tt]==-1 && e[i].cap>0)
            {
                level[tt]=level[u]+1 ;
                if(tt==t)return true ;
                q[rear++]=tt ;
            }
        }
    }
    return false ;
}
int find(int s,int t,int flow)
{
    if(s==t)return flow ;
    int ret=0,a ;
    for(int i=head[s] ;i!=-1 ;i=e[i].nxt)
    {
        int tt=e[i].t ;
        if(level[tt]==level[s]+1 && e[i].cap>0)
        {
            a=find(tt,t,min(e[i].cap,flow-ret)) ;
            e[i].cap-=a ;
            e[i^1].cap+=a ;
            ret+=a ;
            if(ret==flow)
                return ret ;
        }
    }
    if(!ret)level[s]=-1 ;
    return ret ;
}
int dinic(int s,int t)
{
    int flow,ret=0 ;
    while(build(s,t))
        while(flow=find(s,t,INF))
            ret+=flow ;
    return ret ;
}
int l[100005] ;
int IN[205],OUT[205] ;
int C[100005] ;
int main()
{
    int t ;
    scanf("%d",&t) ;
    while(t--)
    {
        scanf("%d%d",&n,&m) ;
        cnt=0 ;
        memset(head,-1,sizeof(head)) ;
        memset(IN,0,sizeof(IN)) ;
        memset(OUT,0,sizeof(OUT)) ;
        for(int i=0 ;i<m ;i++)
        {
            int a,b,w ;
            scanf("%d%d%d%d",&a,&b,&l[i],&w) ;
            IN[b]+=l[i] ;OUT[a]+=l[i] ;
            C[i]=w ;
            add(a,b,w-l[i]) ;
        }
        int S,T ;
        S=0 ;T=n+1 ;
        int sum=0 ;
        for(int i=1 ;i<=n ;i++)
        {
            int f=IN[i]-OUT[i] ;
            if(f>0)
            {
                add(S,i,f) ;
                sum+=f ;
            }
            else
                add(i,T,-f) ;
        }
        if(dinic(S,T)==sum)
        {
            puts("YES") ;
            for(int i=0 ;i<m ;i++)
            {
                printf("%d\n",C[i]-e[i*2].cap) ;
            }
        }
        else puts("NO") ;
    }
    return 0 ;
}