UVA 11082 矩阵解压 (网络流）

题目链接

题意很好懂，这题的关键是看出每个所有行的元素总和等于所有列的元素总和，看出这个才能用网络流模板做题，之所以每行，每列-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;
}

}