【分析】
这题是【BZOJ1070修车】的加强版。
若仍正常建边的话,会T。那就想如何优化。
费用流中每次增广,只增广一条最短的道路,及对应一名厨师做一道菜。而每次又找的最短的,所以倒数第k次一定比倒数第k+1次先找到。得到了这个信息,就可以动态加边来解决了。
【建图】
- S向每种菜品连边,流量为pi,费用为0。
- 令P=∑pi,将厨师拆为m*P个点,分别向T连边,流量为1,费用为0。
- 最开始将每个厨师做倒数第一道菜的边加入。假设做第i种菜,第j名厨师。则费用为t[i][j]。流量均为1。
- 每次增广后,假设此次是第j名厨师做的倒数第k道菜。则加入该名厨师做倒数第k+1道菜的边。
具体可以看代码。
【代码】
#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 80045
#define M 320285
#define INF 1000000000
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;
int read()
{
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,m,cnt=1,S,T,ans,sum;
int b[M],p[N],nextedge[M],w[M],c[M];
int Dis[N],Pre[N],t[45][105];
bool Flag[N];
void Add(int x,int y,int z,int cost)
{
cnt++;
b[cnt]=y;
nextedge[cnt]=p[x];
p[x]=cnt;
w[cnt]=z;
c[cnt]=cost;
}
void Anode(int x,int y,int z,int cost){
Add(x,y,z,cost);Add(y,x,0,-cost);
}
void Input_Init()
{
n=read(),m=read();
static int x;
for(int i=1;i<=n;i++)
{
x=read();sum+=x;
Anode(0,i,x,0);
}
T=m*sum+n+1;
for(int i=1;i<=m;i++)
for(int j=1;j<=sum;j++) Anode((i-1)*sum+j+n,T,1,0);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
t[i][j]=read();
Anode(i,(j-1)*sum+1+n,1,t[i][j]);
}
}
bool Bfs()
{
queue<int>q;
q.push(S);
for(int i=1;i<=T;i++) Dis[i]=INF;
while(!q.empty())
{
int k=q.front();q.pop();Flag[k]=0;
for(int i=p[k];i;i=nextedge[i])
{
int v=b[i],f=w[i];
if(Dis[v]>Dis[k]+c[i]&&f)
{
Dis[v]=Dis[k]+c[i];
Pre[v]=i;
if(!Flag[v])
{
Flag[v]=1;
q.push(v);
}
}
}
}
return Dis[T]!=INF;
}
void Mcf()
{
int Maxf=INF,a;
for(int i=Pre[T];i;i=Pre[b[i^1]])
Maxf=min(Maxf,w[i]);
for(int i=Pre[T];i;i=Pre[b[i^1]])
{
w[i]-=Maxf;w[i^1]+=Maxf;
ans+=Maxf*c[i];
}
int v=b[Pre[T]^1];
a=(v-n)/sum+1;v=(v-n)%sum;
for(int i=1;i<=n;i++)
Anode(i,(a-1)*sum+v+1+n,1,t[i][a]*(v+1));
}
void MCF()
{
while(Bfs())
Mcf();
printf("%d\n",ans);
}
int main()
{
Input_Init();
MCF();
return 0;
}