uva11082矩阵解压

题意：

对于一个R行C列的正整数矩阵，(1<=R,C<=20),设a[i]为前i行所有元素之和，b[i]为前i列所有元素之和，已知道R，C，和数组a，b，找一个满足条件的矩阵，矩阵中的元素必须是1~20之间的正整数，输入保证有解。

思路：转化为网络流问题

根据输入先算出每行，每列的和。每行的和减列数c，每列的和减行数r，这是为后面的容量做铺垫。每行（列）视为一个点，每行到每列连一条弧，容量为19（原先最大为20，刚才减了1，所以最大为19），意思是flow(i,j)最大为19，可以想到flow(i,j)就是对应于矩阵中的元素(i,j)的值。
建立源点和汇点，求最大流，最终每条弧的流量+1就对应矩阵中的元素。关键就在于这个转化。

#include <cstdio>
#include <queue>
#include <vector>
#include <cstring>
#include <algorithm>
#define fi first
#define se second
#define pii pair<int,int>
using namespace std;
typedef long long LL;
const int INF = 0x3f3f3f3f; 
const int maxn = 50+5;
int n,m;
int Ai[maxn], Ci[maxn];
int A[maxn], C[maxn];
int Cap[maxn][maxn], flow[maxn][maxn], dis[maxn];

bool bfs(int s, int t){
	memset(dis, -1, sizeof(dis));
	dis[s] = 0;
	queue<int> Q;
	Q.push(s);
	while(!Q.empty()){
		int x = Q.front(); Q.pop();
		for(int i = 0; i < n+m+2; ++i){
			if(dis[i] == -1&&Cap[x][i] > flow[x][i]){
				dis[i] = dis[x] + 1;
				Q.push(i);
			}
		}
	}
	return dis[t] != -1;
}

int dfs(int s, int t, int f){
	if(s == t||f == 0) return f;
	int ans = 0;
	for(int i = 0; i < n+m+2; ++i){
		if(dis[i] == dis[s] + 1&&Cap[s][i] > flow[s][i]){
			int a1 = min(f, Cap[s][i] - flow[s][i]);
			int a2 = dfs(i, t, a1);
			if(a2 == 0) continue;
			flow[s][i] += a2;
			flow[i][s] -= a2;
			ans += a2;
			f -= a2;
			if(f <= 0) break; 
		}
	}
	return ans;
} 

int dinic(int s, int t){
	int ans = 0;
	while(bfs(s, t)){
		//printf("hh");
		ans += dfs(s, t, INF);
	}
	for(int i = 0; i < n; ++i){
		for(int j = n; j < n+m-1; ++j){
			printf("%d ",flow[i][j]+1);
		}printf("%d\n",flow[i][n+m-1]+1);
	}
	return ans;
}

int main()
{
	//freopen("in.txt","r",stdin);
	int T; scanf("%d",&T);
	for(int kase = 1; kase <= T; ++kase){
		scanf("%d%d",&n,&m);
		// 算出每行每列的和 
		scanf("%d",&Ai[0]); A[0] = Ai[0];
		for(int i = 1; i < n; ++i) {
			scanf("%d",&Ai[i]);
			A[i] = Ai[i] - Ai[i-1]; 
		}
		
		scanf("%d",&Ci[0]); C[0] = Ci[0];
		for(int i = 1; i < m; ++i) {
			scanf("%d",&Ci[i]);
			C[i] = Ci[i] - Ci[i-1];
		}
		
		// 建立网络 
		memset(Cap, 0, sizeof(Cap));
		memset(flow, 0, sizeof(flow));
		for(int i = 0; i < n; ++i) A[i] -= m;
		for(int i = 0; i < m; ++i) C[i] -= n;
		
		//for(int i = 0; i < n; ++i) printf("%d ",A[i]);
		//for(int i = 0; i < m; ++i) printf("%d ",C[i]);
		
		for(int i = 0; i < n; ++i)
		for(int j = n; j < n+m; ++j) Cap[i][j] = 19;
		for(int i = 0; i < n; ++i) Cap[n+m][i] = A[i];
		for(int i = n; i < n+m; ++i) Cap[i][n+m+1] = C[i-n];
		
		printf("Matrix %d\n",kase);
		
		int ans = dinic(n+m, n+m+1);
		//printf("%d\n", ans);
		if(kase < T) printf("\n");
	}

	return 0;
}