uva 11082

时间:2023-03-09 04:06:28
uva 11082

题意:知道矩阵的前i行之和,和前j列之和(任意i和j都可以)。求这个矩阵。每个格子中的元素必须在1~20之间。矩阵大小上限20*20

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#define pb push_back
using namespace std;
const int INF=0x3f3f3f3f;
int ans[][],R,C,r[],c[],A[],B[];
struct EdmondsKarp
{
struct Edge
{
int from,to,cap,flow;
};
vector<Edge>edges;
vector<int>G[];
int p[],a[];
void init()
{
for(int i=;i<;i++)G[i].clear();
edges.clear();
memset(p,,sizeof(p));
}
void addedge(int from,int to,int cap)
{
edges.pb((Edge){from,to,cap,});
edges.pb((Edge){to,from,,});
int m=edges.size();
G[from].pb(m-);
G[to].pb(m-);
}
int maxflow(int s,int t)
{
int flow=;
while()
{
memset(a,,sizeof(a));
queue<int>q;
q.push(s);
a[s]=INF;
while(!q.empty())
{
int x=q.front();
q.pop();
for(int i=;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]^].flow-=a[t];
}
flow+=a[t];
}
return flow;
}
void solve()
{
int temp=maxflow(,R+C+);
for(int len=edges.size(),i=;i<len;i+=)
{
Edge& e=edges[i];
if(e.from!=&&e.to!=R+C+)
ans[e.from][e.to-R]=e.flow+;
}
for(int i=;i<=R;i++)
for(int j=;j<=C;j++)
printf(j==C?"%d\n":"%d ",ans[i][j]);
}
}EK;
int main()
{
int T;
scanf("%d",&T);
bool flag=;
for(int kase=;kase<=T;kase++)
{
scanf("%d%d",&R,&C);
EK.init();
for(int i=;i<=R;i++)scanf("%d",&r[i]);
for(int i=;i<=R;i++)A[i]=r[i]-r[i-];
for(int i=;i<=C;i++)scanf("%d",&c[i]);
for(int i=;i<=C;i++)B[i]=c[i]-c[i-];
for(int i=;i<=R;i++)
EK.addedge(,i,A[i]-C);
for(int i=;i<=C;i++)
EK.addedge(i+R,R+C+,B[i]-R);
for(int i=;i<=R;i++)
for(int j=;j<=C;j++)
EK.addedge(i,R+j,);
if(kase>)putchar('\n');
printf("Matrix %d\n",kase);
EK.solve();
}
return ;
}