POJ - 2396 Budget（上下界网络流）

Budget 题目链接
题意：
有$n\ast m$的数字矩阵，告诉你每行和每列的数字之和，接下来有$c$行约束条件，第$c$行会告诉你第$i$行第$j$列的格子应该有怎样的约束$(>,<,=)$，请输出任一种满足所有约束的矩阵或者输出不可能
思路：
经典的建图模型：
将每行每列都抽象为$1$个点，源点与行相连，汇点与列相连，流量上下界均为$su{m}_l/su{m}_c$
每行每列直接的连线流量代表这些位置的约束范围
首先要保证所有的约束不会矛盾，否则无解。
建好图后在图上判断是否有可行流
最后根据图上每个位置流出的流量即可计算每个点可以的数字

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#define inf 0x3f3f3f3f
#define double long double
using namespace std;
const int maxn=255;
int depth[maxn],now[maxn],num,head[maxn],s,t,n,m,ok,up[maxn][maxn],low[maxn][maxn],in[maxn],out[maxn];
int res[maxn][maxn];
struct road{int b,c,nex;}r[2000005];
void add(int a,int b,int c){r[num].b=b;r[num].c=c;r[num].nex=head[a];head[a]=num++;}
void addedge(int a,int b,char op,int val)
{
	if(op=='=')
	{
		if(val<low[a][b] || val>up[a][b]) ok=0;
		up[a][b]=low[a][b]=val;
	}
	else if(op=='>')
	{
		if(val>=up[a][b]) ok=0;
		low[a][b]=max(low[a][b],val+1);
	}
	else
	{
		if(val<=low[a][b]) ok=0;
		up[a][b]=min(up[a][b],val-1);
	}
}
int make_level()
{
	memset(depth,-1,sizeof(depth));
	queue<int>q;
	q.push(s);
	depth[s]=1;
    now[s]=head[s];
	while(!q.empty())
	{	
		int u=q.front();q.pop();
		for(int i=head[u];~i;i=r[i].nex)
		{
			int v=r[i].b;
			if(depth[v]!=-1 || r[i].c<=0) continue;
            now[v]=head[v];
			depth[v]=depth[u]+1;
			q.push(v);
		}
	}
	return depth[t]!=-1;
}
int dinic(int u,int flow)
{
	if(u==t) return flow;
	int sum=0;
	for(int i=now[u];~i;i=r[i].nex)
	{
        now[u]=i;
		int v=r[i].b;
		if(depth[v]!=depth[u]+1 || r[i].c<=0) continue;
		int use=dinic(v,min(flow-sum,r[i].c));
		if(use)
		{
			r[i].c-=use;
			r[i^1].c+=use;
			sum+=use;
		}
		if(sum==flow) return flow;
	}
	if(sum==0) depth[u]=-1;
	return sum;
}
int main()
{
	int T;scanf("%d",&T);
	while(T--)
	{
		memset(head,-1,sizeof(head));num=0;ok=1;
		memset(low,0,sizeof(low));
		memset(up,0x3f,sizeof(up));
		memset(in,0,sizeof(in));
		memset(out,0,sizeof(out));
		scanf("%d%d",&n,&m);
		s=m+n+1,t=s+1;
		for(int i=1;i<=n;i++)
		{
			int x;scanf("%d",&x);
			out[s]+=x;in[i]+=x;
		}
		for(int i=1;i<=m;i++)
		{
			int x;scanf("%d",&x);
			out[i+n]+=x;in[t]+=x;
		}
		int k;scanf("%d",&k);
		while(k--)
		{
			int a,b,v;char op[2];
			scanf("%d%d%s%d",&a,&b,op,&v);
			if(a&&b)
			{
				addedge(a,b,op[0],v);
			}
			else if(a)
			{
				for(int i=1;i<=m;i++) addedge(a,i,op[0],v);
			}
			else if(b)
			{
				for(int i=1;i<=n;i++) addedge(i,b,op[0],v);
			}
			else
			{
				for(int i=1;i<=n;i++)
				{
					for(int j=1;j<=m;j++)
					{
						addedge(i,j,op[0],v);
					}
				}
			}
		}
		if(!ok)
		{
			puts("IMPOSSIBLE");continue;
		}
		int s2=t+1,t2=s2+1,sum=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				int ff=up[i][j]-low[i][j];
				add(i,j+n,ff);add(j+n,i,0);
				out[i]+=low[i][j];in[j+n]+=low[i][j];
			}
		}
		for(int i=1;i<=n+m+2;i++)
		{
			if(out[i]>in[i]) sum+=(out[i]-in[i]),add(i,t2,(out[i]-in[i])),add(t2,i,0);
			else if(in[i]>out[i]) add(s2,i,in[i]-out[i]),add(i,s2,0);
		}
		add(t,s,inf);add(s,t,0);
		s=s2;t=t2;
		int ans=0;
		while(make_level()) ans+=dinic(s,inf);
		if(ans!=sum)
		{
			puts("IMPOSSIBLE");continue;
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=head[i];~j;j=r[j].nex)
			{
				int nx=r[j].b-n;
				if(nx<1 || nx>m) continue;
				res[i][nx]=low[i][nx]+r[1^j].c;
			}
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				printf("%d ",res[i][j]);
			}
			puts("");
		}
	}
}