美食节

动态加点的费用流

#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;
}