简介:
给出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;
}