BZOJ3789 : 扫雪车

有上下界的网络流

T向S连容量为正无穷的边，将有源汇转化为无源汇

每条边容量减去下界，设in[i]表示流入i的下界之和减去流出i的下界之和

新建超级源汇SS,TT

对于in[i]>0的点，SS向i连容量为in[i]的边

对于in[i]<0的点，i向TT连容量为-in[i]的边

求出以SS,TT为源汇的最大流，如果等于$\sum in[i](in[i]>0)$，则有解

再求出以S,T为源汇的最大流即为答案

 

#include<cstdio>
const int N=110,inf=~0U>>2;
int n,m,i,j,w,t,S,T,SS,TT,h[N],gap[N],maxflow,sum,in[N],id[N];
struct edge{int t,f;edge *nxt,*pair;}*g[N],*d[N];
inline int min(int a,int b){return a<b?a:b;}
inline void swap(int&a,int&b){int c=a;a=b;b=c;}
inline void add(int s,int t,int f){
  edge *p=new(edge);p->t=t;p->f=f;p->nxt=g[s];g[s]=p;
  p=new(edge);p->t=s;p->f=0;p->nxt=g[t];
  g[t]=p;g[s]->pair=g[t];g[t]->pair=g[s];
}
int sap(int v,int flow,int S,int T){
  if(v==T)return flow;
  int rec=0;
  for(edge *p=d[v];p;p=p->nxt)if(h[v]==h[p->t]+1&&p->f){
    int ret=sap(p->t,min(flow-rec,p->f),S,T);
    p->f-=ret;p->pair->f+=ret;d[v]=p;
    if((rec+=ret)==flow)return flow;
  }
  d[v]=g[v];
  if(!(--gap[h[v]]))h[S]=TT;
  gap[++h[v]]++;
  return rec;
}
int main(){
  scanf("%d%d%d%d",&n,&m,&S,&T);
  for(i=1;i<=n;i++)id[i]=i;
  swap(id[S],id[n-1]),swap(id[T],id[n]);
  S=n-1,T=S+1,SS=T+1,TT=SS+1;add(T,S,inf);
  while(m--){
    scanf("%d%d%d%d",&i,&j,&w,&t);
    i=id[i],j=id[j];
    if(t)in[i]-=w,in[j]+=w;else add(i,j,w);
  }
  for(i=1;i<=TT;i++)if(in[i]>0)sum+=in[i],add(SS,i,in[i]);else add(i,TT,-in[i]);
  for(gap[i=0]=TT;i++<TT;)d[i]=g[i];
  while(h[SS]<TT)maxflow+=sap(SS,inf,SS,TT);
  if(sum!=maxflow)return puts("0"),0;
  for(maxflow=i=0;i<=TT;i++)d[i]=g[i],h[i]=gap[i]=0;
  gap[0]=TT;
  while(h[S]<TT)maxflow+=sap(S,inf,S,T);
  return printf("%d",maxflow),0;
}