题目:UVA11082
题意:有一个n行m列的矩阵,给出前n行的和,前m列的和,让构造出这个矩阵,保证有解,矩阵中的每个值在1-20之间
源点向每一行连边,每一列向汇点连边,然后每个格子就是该行与该列连边,题目中说每一个矩阵的小格子的值必须大于0,如果这时候我们把容量设为20是不对的,因为这样会出现跑出来0的情况。所以对于行与列之间的连边,容量应设为19,也就是我们默认流了1的流量了,所以源点向每一行连的边的容量为这一行的和减去列数,因为每一行有列数个格子嘛,每个格子有1,列与汇点的连边同理,跑一遍最大流以后输出时每个格子再加1
输出的时候可把我烦死了QAQ
#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<vector>
#include<stack>
using namespace std;
const int maxn=1010;
const int maxm=1e6+7;
const int inf=0x3f3f3f3f;
struct Node
{
int from;
int to;
int capa;
int flow;
int next;
}edge[maxm];
int n,m;
int cnt;
int head[maxn];
int source,sink;
int numR[maxn];
int numC[maxn];
int dep[maxn];
int map[110][110];
void init()
{
cnt=0;
memset(head,-1,sizeof(head));
return;
}
void add(int u,int v,int capa)
{
edge[cnt].from=u;
edge[cnt].to=v;
edge[cnt].capa=capa;
edge[cnt].flow=0;
edge[cnt].next=head[u];
head[u]=cnt++;
edge[cnt].from=v;
edge[cnt].to=u;
edge[cnt].capa=0;
edge[cnt].flow=0;
edge[cnt].next=head[v];
head[v]=cnt++;
return;
}
bool bfs()
{
memset(dep,-1,sizeof(dep));
queue<int> que;
que.push(source);
dep[source]=0;
while(!que.empty())
{
int node=que.front();
que.pop();
for(int i=head[node];~i;i=edge[i].next)
{
int v=edge[i].to;
if(dep[v]==-1&&edge[i].capa>edge[i].flow)
{
dep[v]=dep[node]+1;
if(v==sink) break;
que.push(v);
}
}
}
return dep[sink]!=-1;
}
int dfs(int node,int minn)
{
if(node==sink||!minn)
{
return minn;
}
int r=0;
for(int i=head[node];~i;i=edge[i].next)
{
int v=edge[i].to;
if(dep[v]==dep[node]+1)
{
int tmp=dfs(v,min(minn,edge[i].capa-edge[i].flow));
if(tmp>0)
{
edge[i].flow+=tmp;
edge[i^1].flow-=tmp;
r+=tmp;
minn-=tmp;
if(!minn) break;
}
}
}
if(!r) dep[node]=-1;
return r;
}
void dinic()
{
int maxflow=0;
while(bfs())
{
maxflow+=dfs(source,inf);
}
return;
//return maxflow;
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int test;
scanf("%d",&test);
for(int cas=1;cas<=test;cas++)
{
vector<int> vec;
init();
scanf("%d%d",&n,&m);
source=0;
sink=n+m+1;
numR[0]=numC[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&numR[i]);
add(source,i,numR[i]-numR[i-1]-m);
}
for(int i=1;i<=m;i++)
{
scanf("%d",&numC[i]);
add(n+i,sink,numC[i]-numC[i-1]-n);
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
add(i,j+n,19);
}
}
dinic();
/*for(int i=15;i<cnt;i++)
{
//vec.push_back(edge[i].flow);
cout<<edge[i].flow<<' ';
}*/
for(int i=1;i<=n;i++)
{
for(int j=head[i];~j;j=edge[j].next)
{
if(edge[j].to-n>=1)
{
vec.push_back(edge[j].flow+1);
}
//cout<<edge[j].from<<' '<<edge[j].to-n<<' '<<edge[j].capa<<' '<<edge[j].flow<<endl;
}
}
printf("Matrix %d\n",cas);
int len = vec.size();
stack<int> sta;
for(int i=0;i<len;i++)
{
if((i+1)%m==0)
{
sta.push(vec[i]);
while(!sta.empty())
{
if(sta.size()==1)
{
printf("%d\n",sta.top());
sta.pop();
}
else
{
printf("%d ",sta.top());
sta.pop();
}
}
//printf("%d\n",vec[i]);
}
else
{
sta.push(vec[i]);
}
//printf("%d ",vec[i]);
}
puts("");
}
return 0;
}
//