Matrix Decompressing UVA - 11082 最大流 + 输出方案

传送门

题意：

给出前$i$行的元素和$A[i]$，前$j$列的元素$B[j]$，让你构造一个矩阵使得其满足前$i$行的元素和是$A[i]$，前$j$列的元素和是$B[j]$。
$r,c\le 20$，矩阵中元素$1\le x\le 20$。

思路：

首先我们知道了前$i$行的元素和以及列的元素和，那么第$i$行或列的元素和我们也就知道了。
首先猜到这是一个网络流的问题，考虑行和列将其看成一个二分图，由于网络流中存在流量为$0$的边，所以将矩阵中元素都减$1$，即矩阵中元素$0\le x\le 19$，接下来源点向每一行连容量为这一行的和的边，每一列向汇点连容量为这一列的和的边，再从每一行向每一列连流量为$19$边，代表$(i,j)$这个点可以从$[0,19]$中选一个值流向汇点。最终有解的话一定是汇点满流，由于总和相等所以此时源点也满流。
最后输出一下方案就好啦。

//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native")
//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=100010,M=N*2,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n,m;
int A[N],B[N],a[N],b[N];
int ans[40][40];
struct Maxflow
{
    int n,m,S,T;
    int e[M],ne[M],w[M],h[N],hs[N],idx;
    int depth[N];

    void init() { idx=0; memset(h,-1,sizeof (h)); }

    inline void add(int a,int b,int c)
    {
        e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
        e[idx]=a,w[idx]=0,ne[idx]=h[b],h[b]=idx++;
    }

    bool bfs()
    {
        memset(depth,-1,sizeof (depth)); depth[S]=0;
        queue<int>q; q.push(S); hs[S]=h[S];
        while(q.size())
        {
            int t=q.front(); q.pop();
            for(int i=h[t];~i;i=ne[i])
            {
                int ver=e[i];
                if(depth[ver]==-1&&w[i]>0)
                {
                    depth[ver]=depth[t]+1;
                    hs[ver]=h[ver];
                    if(ver==T) return true;
                    q.push(ver);
                }
            }
        }
        return false;
    }

    int dfs(int u,int flow)
    {
        if(u==T) return flow;
        int d=flow;
        for(int i=hs[u];~i;i=ne[i])
        {
            int ver=e[i]; hs[u]=i;
            if(depth[ver]==depth[u]+1&&w[i]>0)
            {
                int t=dfs(ver,min(w[i],d));
                if(!t) depth[ver]=-1;
                d-=t; w[i]-=t; w[i^1]+=t;
                if(!d) break;
            }
        }
        return flow-d;
    }

    int dinic()
    {
        int ans=0,flow;
        while(bfs()) while(flow=dfs(S,INF)) ans+=flow;
        return ans;
    }
}MF;

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

	int _; scanf("%d",&_);
	int cnt=0;
	while(_--) {
		cnt++;
		printf("Matrix %d\n",cnt);
		MF.init();
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++) scanf("%d",&A[i]),a[i]=A[i]-A[i-1];
		for(int i=1;i<=m;i++) scanf("%d",&B[i]),b[i]=B[i]-B[i-1];
		MF.S=N-1; MF.T=N-2;
		for(int i=1;i<=n;i++) MF.add(MF.S,i,a[i]-m);
		for(int i=1;i<=m;i++) MF.add(i+n,MF.T,b[i]-n);
		for(int i=1;i<=n;i++) 
			for(int j=1;j<=m;j++) 
				MF.add(i,j+n,19);
		int an=MF.dinic();
		for(int i=0;i<MF.idx;i++) {
			if(MF.e[i^1]>=1&&MF.e[i^1]<=n&&MF.e[i]>n&&MF.e[i]<=n+m) 
				ans[MF.e[i^1]][MF.e[i]-n]=(19-MF.w[i])+1;
		}
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=m;j++) printf("%d ",ans[i][j]);
			puts("");
		}
		puts("");
	}
	



	return 0;
}
/*

*/