一开始想拆点,用n*k个点和m*k个点匹配,求最小花费
然后就是5000个点的最小费用最大流算法,妥妥的超时了
看了别人写的,可以把每种商品分开单独计算,每次就n+m个点的图,算k次加起来
写的时候有个地方没注意,直接break,导致输入偏差了,WA了好多次
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=200+10;
const int M=20000+10;
const int INF=0x7f7f7f7f;
struct Edge
{
int to,nxt,cap,flow,cost;
}edge[M];
int tot,first[N];
void addedge(int u,int v,int w,int rw,int cost)
{
edge[tot].to=v;edge[tot].cap=w;edge[tot].flow=0;
edge[tot].cost=cost;edge[tot].nxt=first[u];first[u]=tot++;
edge[tot].to=u;edge[tot].cap=rw;edge[tot].flow=0;
edge[tot].cost=-cost;edge[tot].nxt=first[v];first[v]=tot++;
}
void init()
{
tot=0;
memset(first,-1,sizeof(first));
}
int pre[N],dis[N];
bool vis[N];
bool spfa(int s,int t,int n)
{
queue<int> q;
for(int i=0;i<n;i++)
{
dis[i]=INF;
pre[i]=-1;
vis[i]=false;
}
q.push(s);
dis[s]=0;
vis[s]=true;
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=false;
for(int i=first[u];i!=-1;i=edge[i].nxt)
{
int v=edge[i].to;
if(edge[i].cap>edge[i].flow&&dis[u]+edge[i].cost<dis[v])
{
dis[v]=dis[u]+edge[i].cost;
pre[v]=i;
if(!vis[v])
{
vis[v]=true;
q.push(v);
}
}
}
}
if(pre[t]==-1) return false;
else return true;
}
int MCMF(int s,int t,int &cost,int n)
{
int flow=0;
cost=0;
while(spfa(s,t,n))
{
int Min=INF;
for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
if(Min>edge[i].cap-edge[i].flow)
Min=edge[i].cap-edge[i].flow;
for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
{
edge[i].flow+=Min;
edge[i^1].flow-=Min;
cost+=edge[i].cost*Min;
}
flow+=Min;
}
return flow;
}
int in[N][N];
int out[N][N];
int main()
{
int n,m,k;
int a;
while(~scanf("%d%d%d",&n,&m,&k)&&(n+m+k))
{
for(int i=1;i<=n;i++)
for(int j=1;j<=k;j++)
scanf("%d",&in[i][j]);
for(int i=1;i<=m;i++)
for(int j=1;j<=k;j++)
scanf("%d",&out[i][j]);
bool flag=true;
int ans=0;
for(int w=1;w<=k;w++)
{
init();
int sum=0,sum0=0;
for(int i=1;i<=n;i++)
addedge(0,i,in[i][w],0,0),sum+=in[i][w];
for(int i=1;i<=m;i++)
addedge(n+i,n+m+1,out[i][w],0,0),sum0+=out[i][w];
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
scanf("%d",&a);
addedge(i,n+j,INF,0,a);
}
if(sum>sum0||!flag)
{
flag=false;
continue; //这里写成了break,WA好多次
}
int cost;
MCMF(0,n+m+1,cost,n+m+2);
ans+=cost;
}
if(flag) printf("%d\n",ans);
else printf("-1\n");
}
return 0;
}