UVa11082 - Matrix Decompressing（最大流）

题目链接

简介：
给出A[i]表示矩阵前i行的元素和
B[i]表示矩阵前i列的元素和
求原始矩阵

分析：
首先我们把A，B数组变成每一行（每一列）的元素和
显然（i，j）既会被计算在A[i]内，又会被计算在B[j]内
这就提示我们把每一行，每一列看做一个点
行点看做是X部，列点看做是Y部

在第i行和第j行之间连边，表示一个点
原点向X部连边，Y部向汇点连边

这道题的容量有一点讲究：
因为题目要求矩阵内的数字必须是1~20，如果按照朴素的建图方式，是会有0流的
那怎么办呢，难道要用有上下界的有汇有源的网络流吗
实际上远远不用这么麻烦
题目表示每个矩阵一定有解，所以每个格子至少是1
也就是说ta们的下界都是一样的，那么我们就可以以1为基准，在这个基础上跑网络流，此时的流量实际上表示增量

所以我们的建图是这样的：

最后输出的时候别忘了+1

//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>

using namespace std;

const int N=101;
const int INF=0x33333333;
int st[N],tot,n,m,s,t;
struct node{
	int x,y,v,nxt;
};
node way[N*N*2];
int cur[N],deep[N];
int h[N],z[N],mp[N][N];

void add(int u,int w,int z)
{
	tot++;
	way[tot].x=u;way[tot].y=w;way[tot].v=z;way[tot].nxt=st[u];st[u]=tot;
	tot++;
	way[tot].x=w;way[tot].y=u;way[tot].v=0;way[tot].nxt=st[w];st[w]=tot;
}

int bfs(int s,int t)
{
	queue<int> Q;
	Q.push(s);
	for (int i=s;i<=t;i++) cur[i]=st[i];
	memset(deep,-1,sizeof(deep));
	deep[s]=1;
	
	while (!Q.empty())
	{
		int now=Q.front(); Q.pop();
		for (int i=st[now];i!=-1;i=way[i].nxt)
		    if (way[i].v&&deep[way[i].y]==-1)
		    {
		    	deep[way[i].y]=deep[now]+1;
		    	Q.push(way[i].y);
			}
	}
	return deep[t]!=-1;
}

int dfs(int now,int t,int limit)
{
	if (now==t||!limit) return limit;
	int f,flow=0;
	for (int i=cur[now];i!=-1;i=way[i].nxt)
	{
		cur[now]=i;
		if (way[i].v&&deep[way[i].y]==deep[now]+1&&(f=dfs(way[i].y,t,min(way[i].v,limit))))
		{
			flow+=f;
			limit-=f;
			way[i].v-=f;
			way[i^1].v+=f;
			if (!limit) break;
		}
	}
	return flow;
}

int dinic(int s,int t)
{
	int ans=0;
	while (bfs(s,t))
	    ans+=dfs(s,t,INF);
	return ans;
}

void build()
{
	int i,j;
	memset(mp,0,sizeof(mp));
	for (i=1;i<=n;i++)
	    for (j=1;j<=m;j++)
	    {
	    	add(i,j+n,19);
	    	mp[i][j]=tot;
		}
	for (i=1;i<=n;i++) add(s,i,h[i]-m);      //-m
	for (i=1;i<=m;i++) add(i+n,t,z[i]-n);    //-n
}

void print()
{
	int i,j;
	for (int i=1;i<=n;i++)
	{
		printf("%d",way[mp[i][1]].v+1);         //+1
		for (int j=2;j<=m;j++)
			printf(" %d",way[mp[i][j]].v+1);    //+1
		printf("\n");
	}
}

int main()
{
	int T,cnt=0;
	scanf("%d",&T);
	for (int cas=1;cas<=T;cas++)
	{
		if (cnt++) printf("\n");
		printf("Matrix %d\n",cas);	
		
		tot=-1;
		memset(st,-1,sizeof(st));
		scanf("%d%d",&n,&m);
		s=0;t=n+m+1;
		
		for (int i=1;i<=n;i++)
	    	scanf("%d",&h[i]);
		for (int i=1;i<=m;i++)
			scanf("%d",&z[i]);
			
		for (int i=n;i>=1;i--) h[i]=h[i]-h[i-1];
		for (int i=m;i>=1;i--) z[i]=z[i]-z[i-1];
		
		build();
		dinic(s,t);
		print();
	}
	return 0;
}