zoj 3229 有源匯有上下界的最大流模板題
/*坑啊,pe的程序在zoj上原來是wa。
題目大意:一個屌絲給m個女神拍照,計劃拍照n天,每一天屌絲最多個C個女神拍照,每天拍照數不能超過D張,而且給每個女神i拍照有數量限制[Li,Ri],
對於每個女神n天的拍照總和不能超過Gi,如果有解求屌絲最多能拍多少張照,並求每天給對應女神拍多少張照;否則輸出-1。
解題思路:增設一源點st,匯點sd,st到第i天連一條上界為Di下界為0的邊,每個女神到匯點連一條下界為Gi上界為oo的邊,對於每一天,當天到第i個女孩連一條[Li,Ri]的邊。
建圖模型:源點s,終點d。超級源點ss,超級終點dd。首先判斷是否存在滿足所有邊上下界的可行流,方法可以轉化成無源匯有上下界的可行流問題。怎麼轉換呢?
增設一條從d到s沒有下界容量為無窮的邊,那麼原圖就變成了一個無源匯的循環流圖。接下來的事情一樣,超級源點ss連i(du[i]>0),i連超級匯點(du[i]<0),
對(ss,dd)進行一次最大流,當maxflow等於所有(du[]>0)之和時,有可行流,否則沒有。
當有可行流時,刪除超級源點ss和超級終點dd,再對(s,d)進行一次最大流,此時得到的maxflow則為題目的解。為什麼呢?
因為第一次maxflow()只是求得所有滿足下界的流量,而殘留網絡(s,d)路上還有許多自由流(沒有和超級源點和超級匯點連接的邊)沒有流滿,
所有最終得到的maxflow=(第一次流滿下界的流+第二次能流通的自由流)。
*/
#include
#include
#include
using namespace std;
#define N 1500
#define ii 400000
#define inf 0x3fffffff
struct node {
int u,v,w,f,next;
}bian[ii*2];
int head[N],yong,dis[N],work[N];
void init(){
yong=0;
memset(head,-1,sizeof(head));
}
void addbian(int u,int v,int w,int f) {
bian[yong].u=u;
bian[yong].v=v;
bian[yong].w=w;
bian[yong].f=f;
bian[yong].next=head[u];
head[u]=yong++;
}
void add(int u,int v,int w,int f) {
addbian(u,v,w,f);
addbian(v,u,0,f);
}
int min(int a,int b)
{
return aq;
q.push(s);
dis[s]=0;
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=head[u];i!=-1;i=bian[i].next)
{
int v=bian[i].v;
if(bian[i].w&&dis[v]==-1)
{
dis[v]=dis[u]+1;
q.push(v);
if(v==t)
return 1;
}
}
}
return 0;
}
int dfs(int s,int limit,int t)
{
if(s==t)return limit;
for(int &i=work[s];i!=-1;i=bian[i].next)
{
int v=bian[i].v;
if(bian[i].w&&dis[v]==dis[s]+1)
{
int tt=dfs(v,min(limit,bian[i].w),t);
if(tt)
{
bian[i].w-=tt;
bian[i^1].w+=tt;
return tt;
}
}
}
return 0;
}
int dinic(int s,int t)
{
int ans=0;
while(bfs(s,t))
{
memcpy(work,head,sizeof(head));
while(int tt=dfs(s,inf,t))
ans+=tt;
}
return ans;
}
int main() {
int n,m,i,j,k,s,t,S,T,a,b,d,index[ii],id,suma,w[N];
while(scanf("%d%d",&n,&m)!=EOF) {
init();
s=0;t=m+n+1;
S=t+1;T=S+1;
memset(w,0,sizeof(w));
for(i=1;i<=m;i++) {
scanf("%d",&a);
w[t]+=a;
w[i+n]-=a;
add(i+n,t,inf-a,0);
}
id=0;
for(i=1;i<=n;i++) {
scanf("%d%d",&a,&b);
add(s,i,b,b);
while(a--) {
scanf("%d%d%d",&j,&k,&d);
j++;
index[id++]=yong;
w[i]-=k;
w[j+n]+=k;
add(i,j+n,d-k,d);
}
}
add(t,s,inf,inf);
suma=0;
for(i=s;i<=t;i++) {
if(w[i]>0) {
suma+=w[i];
add(S,i,w[i],0);
}
else
add(i,T,-w[i],0);
}
int f=dinic(S,T);
if(f==suma) {
head[S]=head[T]=-1;
printf("%d\n",dinic(s,t));
for(i=0;i