之前还自以为全部塞在一个图里面就可以搞了。谁知,这样不行。
分开搞网络流,一方面可以节约某些时间,一方面有时候会简化问题。
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
#define MAXPT 126000
#define MAXN 50
#define INF 2000000000
struct node{
int v,w,c;
node *next,*back;
}edge[MAXPT*4+10],*adj[MAXPT+10],*ecnt=&edge[0],*fa[MAXPT+10];
int n,m,k,need[MAXN+10][MAXN+10],have[MAXN+10][MAXN+10],cost[MAXN+10][MAXN+10][MAXN+10];
int s,t,dist[MAXPT+10],ans,tot;
bool vis[MAXPT+10];
inline void Init()
{
tot=0;
ecnt=&edge[0];
//memset(edge,0,sizeof edge); //加上这个瞬间慢成什么了。。
memset(adj,0,sizeof adj);
}
void addedge(int u,int v,int c,int w)
{
node *p=++ecnt;
p->v=v;
p->c=c,p->w=w;
p->next=adj[u],p->back=ecnt+1;
adj[u]=p;
p=++ecnt;
p->v=u;
p->c=0,p->w=0;
p->next=adj[v],p->back=ecnt-1;
adj[v]=p;
}
void read()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=k;j++)
scanf("%d",&need[i][j]);
for(int i=1;i<=m;i++)
for(int j=1;j<=k;j++)
scanf("%d",&have[i][j]);
for(int p=1;p<=k;p++)
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%d",&cost[p][i][j]);
}
bool SPFA()
{
int u,v;
memset(vis,0,sizeof vis);
for(int i=s;i<=t;i++)
dist[i]=INF;
queue<int> que;
que.push(s);
vis[s]=true,dist[s]=0;
while(!que.empty()){
u=que.front(); que.pop();
vis[u]=false;
for(node *p=adj[u];p;p=p->next){
v=p->v;
if(p->c&&dist[v]>dist[u]+p->w){
dist[v]=dist[u]+p->w;
fa[v]=p;
if(!vis[v]){
que.push(v);
vis[v]=true;
}
}
}
}
if(dist[t]==INF) return false;
else return true;
}
bool mincost_maxflow()
{
int flow=0;
while(SPFA()){
int delta=INF;
node *p;
for(int u=t;u!=s;u=p->back->v){
p=fa[u];
delta=min(delta,p->c);
}
flow+=delta;
for(int u=t;u!=s;u=p->back->v){
p=fa[u];
p->c-=delta;
p->back->c+=delta;
p->back->w=-p->w;
ans+=delta*p->w;
}
}
if(tot==flow) return true;
else return false;
}
bool workout(int kth)
{
Init();
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
addedge(i,n+j,INF,cost[kth][i][j]);
s=0,t=n+m+1;
for(int i=1;i<=n;i++){
addedge(s,i,need[i][kth],0);
tot+=need[i][kth];
}
for(int i=1;i<=m;i++)
addedge(n+i,t,have[i][kth],0);
return mincost_maxflow();
}
int main()
{
while(scanf("%d%d%d",&n,&m,&k)){
if(!n&&!m&&!k) break;
read();
ans=0;
for(int i=1;i<=k;i++)
if(!workout(i)){
ans=-1;
break;
}
printf("%d\n",ans);
}
}