BNUOJ19500 Matrix Decompressing

Posted MashiroSky

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了BNUOJ19500 Matrix Decompressing相关的知识,希望对你有一定的参考价值。

https://www.bnuoj.com/v3/problem_show.php?pid=19500 (题目链接)

题意

  给出一个R行C列的正整数矩阵,设前${A_i}$项为其前i行所有元素之和,${B_i}$项为其前i列所有元素之和,已知R,C,A,B,找出一个满足条件的矩阵。其中每个元素都是1~20的正整数。

Solution

  看到这类矩阵形的题目,首先就要考虑构造二分图,左集代表行,右集代表列,其中每一条边代表一个点。

  这样构完图后,我们发现可以使用有上下界的网络流来解决这个问题。添加源点向左集连一条边上界为${A_i}$,下界也为${A_i}$;右集上的每个点向额外添加的汇点连一条上界为${B_i}$,下界为${B_i}$的边;左集右集之间每两个点连一条上界为20,下界1的边。

  码了一半,发现其实并不需要上下界,直接将每个元素-1,下界就成了0,因为行之和与列之和值相等的,所以可行流即为最大流。

细节

  邻接表的下标从1开始。

代码

// bnuoj19500
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<queue>
#define LL long long
#define inf 2147483640
#define MOD 10000
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;
 
const int maxn=1010;
struct edge {int to,next,w;}e[maxn<<1];
int head[maxn],d[maxn],r[maxn],c[maxn],a[maxn][maxn];
int n,m,cnt=1,ans;

void link(int u,int v,int w) {
	e[++cnt]=(edge){v,head[u],w};head[u]=cnt;
	e[++cnt]=(edge){u,head[v],0};head[v]=cnt;
}
void Init() {
	memset(head,0,sizeof(head));
	cnt=1;ans=0;
}
bool bfs() {
	memset(d,-1,sizeof(d));
	queue<int> q;q.push(1);d[1]=0;
	while (!q.empty()) {
		int x=q.front();q.pop();
		for (int i=head[x];i;i=e[i].next) if (e[i].w && d[e[i].to]<0) {
				d[e[i].to]=d[x]+1;
				q.push(e[i].to);
			}
	}
	return d[n+m+2]>0;
}
int dfs(int x,int f) {
	if (x==n+m+2 || f==0) return f;
	int w,used=0;
	for (int i=head[x];i;i=e[i].next) if (e[i].w && d[e[i].to]==d[x]+1) {
			w=dfs(e[i].to,min(e[i].w,f-used));
			used+=w;
			e[i].w-=w;e[i^1].w+=w;
			if (used==f) return used;
		}
	if (!used) d[x]=-1;
	return used;
}
void Dinic() {
	while (bfs()) ans+=dfs(1,inf);
}
int main() {
	int T;scanf("%d",&T);
	for (int Case=1;Case<=T;Case++) {
		Init();
		scanf("%d%d",&n,&m);
		for (int i=1;i<=n;i++) scanf("%d",&r[i]);
		for (int i=1;i<=m;i++) scanf("%d",&c[i]);
		for (int i=n;i;i--) r[i]-=r[i-1];
		for (int i=m;i;i--) c[i]-=c[i-1];
		for (int i=1;i<=n;i++) link(1,i+1,r[i]-m);
		for (int j=1;j<=m;j++) link(j+n+1,n+m+2,c[j]-n);
		for (int i=1;i<=n;i++)
			for (int j=1;j<=m;j++) link(i+1,j+n+1,19);
		Dinic();
		printf("Matrix %d\n",Case);
		for (int i=1;i<=n;i++)
			for (int j=head[i+1];j;j=e[j].next)
				if (e[j].to>i) a[i][e[j].to-n-1]=20-e[j].w;
		for (int i=1;i<=n;i++) {
			for (int j=1;j<=m;j++) printf("%d ",a[i][j]);
			puts("");
		}
		if (Case<T) puts("");
	}
	return 0;
}

  

以上是关于BNUOJ19500 Matrix Decompressing的主要内容,如果未能解决你的问题,请参考以下文章

BNUOJ 34982 Beautiful Garden

字符串BNUOJ 52781 Book Borders

C1. 组队活动 Small(BNUOJ)

A1. 道路修建 Small(BNUOJ)

BNUOJ 33535 Final Exam Arrangement

BNUOJ 52517 Another Server