题意很好懂,这题的关键是看出每个所有行的元素总和等于所有列的元素总和,看出这个才能用网络流模板做题,之所以每行,每列-1,因为网络流中允许0流的出现,所以把1-20变为0-19,最后把答案+1即可。
建好二分图以后,根据每行对所有列都有贡献,所以把每行对所有列连接起来,接下来求最大流,最大流实现后,(i,j)就是xi到yi的流量。
#include<bits/stdc++.h>
#define maxn 45
#define inf 0x3f
using namespace std;
struct Edge{
int from,to,cap,flow;
Edge(int u,int v,int c,int f):from(u),to(v),cap©,flow(f){}
};
int r, c;
int a[maxn],b[maxn];
int x[maxn],y[maxn];
int map1[21][21];
int sumr[21],sumc[21],vr[21],vc[21];
struct EmondsKarp{
int n,m;
vector edges;
vector G[maxn];
int a[maxn]; //起点到i的可改进量
int p[maxn]; //最短路树上的p的入弧编号
void init(int n)
{
for(int i=0;i<n;i++) G[i].clear();
edges.clear();
}
void addedge(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);
}
int maxflow(int s,int t){
int flow=0;
for(; ; )
{
memset(a,0,sizeof(a));
queue Q;
Q.push(s);
a[s]=inf;
while(!Q.empty()){
int x=Q.front(); Q.pop();
for(int i=0;i<G[x].size();i++){
Edge& e=edges[G[x][i]];
if(!a[e.to] && e.cap>e.flow)
{
p[e.to]=G[x][i];
a[e.to]=min(a[x],e.cap-e.flow);
Q.push(e.to);
}
}
if(a[t]) break;
}
if(!a[t]) break;
for(int u=t;u!=s;u=edges[p[u]].from){
edges[p[u]].flow+=a[t];
edges[p[u]^1].flow-=a[t];
}
flow+=a[t];
}
return flow;
}
};
EmondsKarp ek;
void init()
{
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
}
void read()
{
scanf("%d %d",&r,&c);
for(int i=0;i<r;i++)
{
scanf("%d",&sumr[i]);
if(i>0)
a[i]=sumr[i]-sumr[i-1]-c;
else
a[i]=sumr[i]-c;
}
for(int i=0;i<c;i++)
{
scanf("%d",&sumc[i]);
if(i>0)
b[i]=sumc[i]-sumc[i-1]-r;
else
b[i]=sumc[i]-r;
}
}
void create()
{
ek.init(maxn);
for(int i=0;i<r;i++)
ek.addedge(r+c,i,a[i]);
for(int i=0;i<c;i++)
ek.addedge(r+i,r+c+1,b[i]);
for(int i=0;i<r;i++)
for(int j=r;j<c+r;j++)
ek.addedge(i,j,19);
}
int main()
{
int kase,n=0;
scanf("%d",&kase);
while(kase–)
{
init();
read();
create();
int ans=ek.maxflow(r+c,r+c+1);
printf(“Matrix %d\n”,++n);
for(int i=0;i<r;i++)
for(int j=1;j<ek.G[i].size();j++)
map1[i][j]=ek.edges[ek.G[i][j]].flow;
for(int i=0;i<r;i++)
{
for(int j=0;j<c;j++)
cout<<map1[i][j+1]+1<<" ";
cout<<endl;
}
cout<<endl;
}
}