题目链接 https://cn.vjudge.net/problem/UVA-11082
【题意】
对于一个R行C列的正整数矩阵(1<=R,C<=20)设A[i]为前i行的所有元素和,B[i]为前i列的所有元素和。已知R,C和数组A,B,找出一个满足条件的矩阵,并且矩阵中的元素必须是1~20之间的正整数,输入保证有解,输出任意解即可
【思路】
将行和列抽象成结点,每行对应一个X结点,每列对应一个Y结点,构造一个二分图,添加源点s和汇点t. 对于每个X结点,从s到X[i]连一条弧,容量为该行之和减去列值,同理从Y[i]到t连一条弧,容量为该列之和减去行值,再对每对结点(X[i],Y[j])从X[i]到Y[j]连一条弧,权值为19,跑最大流. 所有s出发的弧和所有到达t的弧均满载说明有解,X[i]到Y[j]上的流量就是原来矩阵中第i行j列的值减一.
#include<bits/stdc++.h>
using namespace std;
const int inf=2e9;
const int maxn=55;
struct Edge{
int from,to,cap,flow;
Edge(int u,int v,int c,int f):from(u),to(v),cap(c),flow(f){}
};
struct ISAP{
int n,m,s,t;//结点数,边数(包括反向弧),源点编号,汇点编号
vector<Edge>edges;
vector<int>g[maxn];
bool vis[maxn];
int d[maxn],cur[maxn];
int p[maxn],num[maxn];
void init(int n){
this->n=n;
for(int i=0;i<=n;++i) g[i].clear();
edges.clear();
}
void add(int from,int to,int cap){
edges.push_back(Edge(from,to,cap,0));
edges.push_back(Edge(to,from,0,0));
m=edges.size();
g[from].push_back(m-2);
g[to].push_back(m-1);
}
bool RevBFS(){
memset(vis,0,sizeof(vis));
queue<int>Q;
Q.push(t);
d[t]=0;
vis[t]=1;
while(!Q.empty()){
int x=Q.front();Q.pop();
for(int i=0;i<g[x].size();i++){
Edge &e =edges[g[x][i]^1];
if(!vis[e.from]&&e.cap>e.flow){
vis[e.from]=1;
d[e.from]=d[x]+1;
Q.push(e.from);
}
}
}
return vis[s];
}
int Augment(){
int x=t, a=inf;
while(x!=s){
Edge &e = edges[p[x]];
a= min(a,e.cap-e.flow);
x=edges[p[x]].from;
}
x=t;
while(x!=s){
edges[p[x]].flow+=a;
edges[p[x]^1].flow-=a;
x=edges[p[x]].from;
}
return a;
}
int Maxflow(int s,int t){
this->s=s,this->t=t;
int flow=0;
RevBFS();
memset(num,0,sizeof(num));
for(int i=0;i<n;i++){
num[d[i]]++;
}
int x=s;
memset(cur,0,sizeof(cur));
while(d[s]<n){
if(x==t){
flow+=Augment();
x=s;
}
int ok=0;
for(int i=cur[x];i<g[x].size();i++){
Edge &e =edges[g[x][i]];
if(e.cap>e.flow&&d[x]==d[e.to]+1){
ok=1;
p[e.to]=g[x][i];
cur[x]=i;
x=e.to;
break;
}
}
if(!ok){
int m=n-1;
for(int i=0;i<g[x].size();i++){
Edge &e =edges[g[x][i]];
if(e.cap>e.flow)
m=min(m,d[e.to]);
}
if(--num[d[x]]==0)
break;
num[d[x]=m+1]++;
cur[x]=0;
if(x!=s)
x=edges[p[x]].from;
}
}
return flow;
}
};
int r,c;
int a[maxn],b[maxn],ans[maxn][maxn];
ISAP solver;
int main(){
int T;
scanf("%d",&T);
for(int kase=1;kase<=T;++kase){
scanf("%d%d",&r,&c);
for(int i=1;i<=r;++i) scanf("%d",&a[i]);
for(int j=1;j<=c;++j) scanf("%d",&b[j]);
solver.init(maxn-1);
for(int i=1;i<=r;++i) solver.add(0,i,(a[i]-a[i-1])-c);
for(int j=1;j<=c;++j) solver.add(20+j,50,(b[j]-b[j-1])-r);
for(int i=1;i<=r;++i){
for(int j=1;j<=c;++j){
solver.add(i,20+j,19);
}
}
solver.Maxflow(0,50);
printf("Matrix %d\n",kase);
for(int i=1;i<=r;++i){
for(int num=0;num<solver.g[i].size();++num){
Edge& e=solver.edges[solver.g[i][num]];
int j=e.to;
ans[i][j-20]=1+e.flow;
}
}
for(int i=1;i<=r;++i){
for(int j=1;j<=c;++j) printf("%d%c",ans[i][j],j==c?'\n':' ');
}
puts("");
}
return 0;
}