【POJ2396】Budget（上下界网络流）

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
usingnamespacestd;
#defineINF0xfffffff
#defineMaxn210
#defineMaxm30

intlim[2][Maxn][Maxm];

structnode
{
intx,y,next,f,o;
}t[Maxn*Maxm*10];intlen;

intfirst[Maxn*Maxn+110],dis[Maxn*Maxm+110];
intst,ed,fst,fed,n,m,sum;
boolok;

intmymax(intx,inty){returnx>y?x:y;}
intmymin(intx,inty){returnx<y?x:y;}

voidins(intx,inty,intf)
{
if(f==0)return;
if(x==fst)sum+=f;
t[++len].x=x;t[len].y=y;t[len].f=f;
t[len].next=first[x];first[x]=len;t[len].o=len+1;
t[++len].x=y;t[len].y=x;t[len].f=0;
t[len].next=first[y];first[y]=len;t[len].o=len-1;
}

voidget_e(intx,inty,intl1,intl2)
{
if(l2<l1){ok=0;return;}
ins(fst,y,l1);ins(x,fed,l1);ins(x,y,l2-l1);
}

voidinit()
{
scanf("%d%d",&n,&m);
memset(first,0,sizeof(first));
len=0;
st=n+m+1;ed=st+1;fst=ed+1,fed=fst+1;
ok=1;sum=0;
for(inti=1;i<=n;i++)
{
intx;
scanf("%d",&x);
get_e(st,i,x,x);
}
for(inti=1;i<=m;i++)
{
intx;
scanf("%d",&x);
get_e(i+n,ed,x,x);
}
intq;
scanf("%d",&q);
chars[10];
for(inti=1;i<=n;i++)
for(intj=1;j<=m;j++)
{
lim[0][i][j]=0;lim[1][i][j]=INF;
}
while(q--)
{
intx,y,z;
scanf("%d%d%s%d",&x,&y,s,&z);
intk1,k2,k3,k4;
if(x==0)k1=1,k2=n;
elsek1=k2=x;
if(y==0)k3=1,k4=m;
elsek3=k4=y;
for(inti=k1;i<=k2;i++)
for(intj=k3;j<=k4;j++)
{
if(s[0]=='>')lim[0][i][j]=mymax(lim[0][i][j],z+1);
if(s[0]=='=')lim[0][i][j]=mymax(lim[0][i][j],z);
if(s[0]=='<')lim[1][i][j]=mymin(lim[1][i][j],z-1);
if(s[0]=='=')lim[1][i][j]=mymin(lim[1][i][j],z);
}
}
for(inti=1;i<=n;i++)
for(intj=1;j<=m;j++)
{
get_e(i,j+n,lim[0][i][j],lim[1][i][j]);
if(ok==0)break;
}
get_e(ed,st,0,INF);
}

queue<int>q;
boolbfs()
{
while(!q.empty())q.pop();
memset(dis,-1,sizeof(dis));
q.push(fst);dis[fst]=0;
while(!q.empty())
{
intx=q.front();q.pop();
for(inti=first[x];i;i=t[i].next)if(t[i].f>0)
{
if(dis[t[i].y]==-1)
{
dis[t[i].y]=dis[x]+1;
q.push(t[i].y);
}
}
}
if(dis[fed]==-1)return0;
return1;
}

intffind(intx,intlow)
{
if(x==fed)returnlow;
intnow=0;
for(inti=first[x];i;i=t[i].next)if(t[i].f>0)
{
inty=t[i].y;
if(now==low)break;
if(dis[y]==dis[x]+1)
{
inta=mymin(t[i].f,low-now);
a=ffind(t[i].y,a);
t[i].f-=a;
t[t[i].o].f+=a;
now+=a;
}
}
if(now==0)dis[x]=-1;
returnnow;
}

intpri[Maxn][Maxm];
voiddinic()
{
intans=0;
while(bfs())
{
ans+=ffind(fst,INF);
}
memset(pri,0,sizeof(pri));
if(ans==sum)
{
for(inti=2;i<=len;i+=2)
{
if(t[i].x>n&&t[i].x<=n+m&&t[i].y<=n+m)
{
pri[t[i].y][t[i].x-n]=t[i].f;
}
}
for(inti=1;i<=n;i++)
{
for(intj=1;j<=m;j++)
printf("%d",pri[i][j]+lim[0][i][j]);
printf("\n");
}
}
elseprintf("IMPOSSIBLE\n");
}

intmain()
{
intT;
scanf("%d",&T);
while(T--)
{
init();
if(ok==0){printf("IMPOSSIBLE\n\n");continue;}
dinic();printf("\n");
}
return0;
}