UVA 11082 Matrix Decompressing (网络流)

题目：UVA11082

题意：有一个n行m列的矩阵，给出前n行的和，前m列的和，让构造出这个矩阵，保证有解，矩阵中的每个值在1-20之间

源点向每一行连边，每一列向汇点连边，然后每个格子就是该行与该列连边，题目中说每一个矩阵的小格子的值必须大于0，如果这时候我们把容量设为20是不对的，因为这样会出现跑出来0的情况。所以对于行与列之间的连边，容量应设为19，也就是我们默认流了1的流量了，所以源点向每一行连的边的容量为这一行的和减去列数，因为每一行有列数个格子嘛，每个格子有1，列与汇点的连边同理，跑一遍最大流以后输出时每个格子再加1

输出的时候可把我烦死了QAQ

#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
using namespace std;
const int maxn=1010;
const int maxm=1e6+7;
const int inf=0x3f3f3f3f;
struct Node
{
	int from;
	int to;
	int capa;
	int flow;
	int next;
}edge[maxm];
int n,m;
int cnt;
int head[maxn];
int source,sink;
int numR[maxn];
int numC[maxn];
int dep[maxn];
int map[110][110];
void init()
{
	cnt=0;
	memset(head,-1,sizeof(head));
	return;
}
void add(int u,int v,int capa)
{
	edge[cnt].from=u;
	edge[cnt].to=v;
	edge[cnt].capa=capa;
	edge[cnt].flow=0;
	edge[cnt].next=head[u];
	head[u]=cnt++;
	edge[cnt].from=v;
	edge[cnt].to=u;
	edge[cnt].capa=0;
	edge[cnt].flow=0;
	edge[cnt].next=head[v];
	head[v]=cnt++;
	return;
}
bool bfs()
{
	memset(dep,-1,sizeof(dep));
	queue<int> que;
	que.push(source);
	dep[source]=0;
	while(!que.empty())
	{
		int node=que.front();
		que.pop();
		for(int i=head[node];~i;i=edge[i].next)
		{
			int v=edge[i].to;
			if(dep[v]==-1&&edge[i].capa>edge[i].flow)
			{
				dep[v]=dep[node]+1;
				if(v==sink) break;
				que.push(v);
			}
		}
	}
	return dep[sink]!=-1;
}
int dfs(int node,int minn)
{
	if(node==sink||!minn)
	{
		return minn;
	}
	int r=0;
	for(int i=head[node];~i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(dep[v]==dep[node]+1)
		{
			int tmp=dfs(v,min(minn,edge[i].capa-edge[i].flow));
			if(tmp>0)
			{
				edge[i].flow+=tmp;
				edge[i^1].flow-=tmp;
				r+=tmp;
				minn-=tmp;
				if(!minn) break;
			}
		}
	}
	if(!r) dep[node]=-1;
	return r;
}
void dinic()
{
	int maxflow=0;
	while(bfs())
	{
		maxflow+=dfs(source,inf);
	}
	return;
	//return maxflow;
}
int main()
{
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int test;
	scanf("%d",&test);
	for(int cas=1;cas<=test;cas++)
	{
		vector<int> vec;
		init();
		scanf("%d%d",&n,&m);
		source=0;
		sink=n+m+1;
		numR[0]=numC[0]=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&numR[i]);
			add(source,i,numR[i]-numR[i-1]-m);
		}
		for(int i=1;i<=m;i++)
		{
			scanf("%d",&numC[i]);
			add(n+i,sink,numC[i]-numC[i-1]-n);
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=m;j++)
			{
				add(i,j+n,19);
			}
		}
		dinic();
		/*for(int i=15;i<cnt;i++)
		{
			//vec.push_back(edge[i].flow);
			cout<<edge[i].flow<<' ';
		}*/
		for(int i=1;i<=n;i++)
		{
			for(int j=head[i];~j;j=edge[j].next)
			{
				if(edge[j].to-n>=1)
				{
					vec.push_back(edge[j].flow+1);
				}
				//cout<<edge[j].from<<' '<<edge[j].to-n<<' '<<edge[j].capa<<' '<<edge[j].flow<<endl;
			}
		}
		printf("Matrix %d\n",cas);
	 	int len = vec.size();
		stack<int> sta;
		for(int i=0;i<len;i++)
		{
			if((i+1)%m==0)
			{
				sta.push(vec[i]);
				while(!sta.empty())
				{
					if(sta.size()==1)
					{
						printf("%d\n",sta.top());
						sta.pop();
					}
					else
					{
						printf("%d ",sta.top());
						sta.pop();
					}
					
				}
				//printf("%d\n",vec[i]);
			}
			else
			{
				sta.push(vec[i]);
			}
			//printf("%d ",vec[i]);
		}
		puts("");
	}
	return 0;
}
//