美食节

动态加点的费用流

#include<iostream>
#include<cstdio>
#include<cstring>
#define inf 0x7fffffff
#define T 100001
using namespace std;
int n,m,tot,cnt=1,ans,t[45][105];
int c[45],d[100005],q[100005],from[100005],head[100005];
bool inq[100005];
struct edge{int from,to,next,c,v;}e[3000005];
void ins(int u,int v,int w,int c)
{
cnt++;
e[cnt].from=u;e[cnt].to=v;
e[cnt].next=head[u];head[u]=cnt;
e[cnt].c=c;e[cnt].v=w;
}
void insert(int u,int v,int w,int c)
{ins(u,v,w,c);ins(v,u,0,-c);}
bool spfa()
{
for(int i=0;i<=T;i++)d[i]=inf;
int t=0,w=1;d[0]=0;inq[0]=1;q[0]=0;
while(t!=w)
{
int now=q[t];t++;if(t==T)t=0;
for(int i=head[now];i;i=e[i].next)
if(e[i].v&&d[e[i].to]>d[now]+e[i].c)
{
d[e[i].to]=d[now]+e[i].c;from[e[i].to]=i;
if(!inq[e[i].to])
{inq[e[i].to]=1;q[w++]=e[i].to;if(w==T)w=0;}
}
inq[now]=0;
}
if(d[T]==inf)return 0;
return 1;
}
void mcf()
{
int x=inf,a,b,y;
for(int i=from[T];i;i=from[e[i].from])
{
x=min(x,e[i].v);
if(e[i].from==0)
{y=e[i].to;a=(y-1)/tot+1;b=y%tot+1;}
}
for(int i=from[T];i;i=from[e[i].from])
{e[i].v-=x;e[i^1].v+=x;ans+=e[i].c*x;}
for(int i=1;i<=m;i++)
insert((a-1)*tot+b,n*tot+i,1,b*t[i][a]);
}
int main()
{
scanf("%d%d",&m,&n);
for(int i=1;i<=m;i++)
{
scanf("%d",&c[i]);
tot+=c[i];
}
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
scanf("%d",&t[i][j]);
for(int i=1;i<=n*tot;i++)
insert(0,i,1,0);
for(int i=1;i<=m;i++)
insert(n*tot+i,T,c[i],0);
for(int i=1;i<=n;i++)
for(int k=1;k<=m;k++)
insert((i-1)*tot+1,n*tot+k,1,t[k][i]);
while(spfa())mcf();
printf("%d",ans);
return 0;
}