P1251 餐巾计划问题
题目描述
一个餐厅在相继的 N N 天里,每天需用的餐巾数不尽相同。假设第 天需要 ri r i 块餐巾( i=1,2,...,N i = 1 , 2 , . . . , N )。餐厅可以购买新的餐巾,每块餐巾的费用为 p p 分;或者把旧餐巾送到快洗部,洗一块需 天,其费用为 f f 分;或者送到慢洗部,洗一块需 天( n>m n > m ),其费用为 s s 分( )。
每天结束时,餐厅必须决定将多少块脏的餐巾送到快洗部,多少块餐巾送到慢洗部,以及多少块保存起来延期送洗。但是每天洗好的餐巾和购买的新餐巾数之和,要满足当天的需求量。
试设计一个算法为餐厅合理地安排好 N N 天中餐巾使用计划,使总的花费最小。编程找出一个最佳餐巾使用计划。
输入格式:
由标准输入提供输入数据。文件第 1 行有 1 个正整数 ,代表要安排餐巾使用计划的天数。
接下来的 N N 行是餐厅在相继的 天里,每天需用的餐巾数。
最后一行包含5个正整数 p,m,f,n,s p , m , f , n , s 。 p p 是每块新餐巾的费用; 是快洗部洗一块餐巾需用天数; f f 是快洗部洗一块餐巾需要的费用; 是慢洗部洗一块餐巾需用天数; s s 是慢洗部洗一块餐巾需要的费用。
输出格式:
将餐厅在相继的 天里使用餐巾的最小总花费输出
输入样例#1:
3
1 7 5
11 2 2 3 1
输出样例#1:
134
说明
N≤2000
N
≤
2000
ri≤10000000
r
i
≤
10000000
p,f,s≤10000
p
,
f
,
s
≤
10000
解:
一个简单的费用流,很容易想到连边。但是这里比较难想的个就是洗餐巾怎么解决?我们的流量已经流向汇点了,总不能还再流回来吧?但是我们想到,从汇点循环回原点,不就相当于再从原点流一个这么多流量出来吗。(但是前提是,一定要满流,不然还没流出去就又流出来啦。)
其实这种方法可以和那种上下界网络流很像(不像?好像是不像)。如果知道事先知道流量,那么我们的操作就变多了,相当于变成一个循环流了。
写了个多路增广的费用流,反正是A了,据说这样是WA的?有没有数据hack一下。反正现在这样写吧,还挺快的。
code:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define inf 1000000000
using namespace std;
struct lxy{
int flow,to,next,w;
}eg[100005];
int N,n,p,m,s,f,ss,tt,cnt=-1;
int head[4005],data[2005],dis[4005],cir[4005];
bool vis[4005];
long long ans;
void add(int op,int ed,int flow,int cost){
eg[++cnt].next=head[op];
eg[cnt].to=ed;
eg[cnt].flow=flow;
eg[cnt].w=cost;
head[op]=cnt;
}
bool spfa(){
queue <int> d;
memset(dis,0x3f3f3f3f,sizeof(dis));
memset(vis,0,sizeof(vis));
d.push(ss),dis[ss]=0;
while(!d.empty()){
int now=d.front();d.pop();vis[now]=0;
for(int i=head[now];i!=-1;i=eg[i].next)
if(eg[i].flow!=0&&dis[eg[i].to]>dis[now]+eg[i].w){
dis[eg[i].to]=dis[now]+eg[i].w;
if(vis[eg[i].to]==0){
vis[eg[i].to]=1;
d.push(eg[i].to);
}
}
}
if(dis[tt]==0x3f3f3f3f) return false;
else return true;
}
int dfs(int u,int a){
if(u==tt||a==0) return a;
vis[u]=1;
int f,flow=0;
for(int &i=cir[u];i!=-1;i=eg[i].next)
if(eg[i].flow!=0&&dis[u]+eg[i].w==dis[eg[i].to]&&vis[eg[i].to]==0){
f=dfs(eg[i].to,min(a,eg[i].flow));
flow+=f;a-=f;
eg[i].flow-=f;eg[i^1].flow+=f;
ans+=1ll*f*eg[i].w;
if(a==0) break;
}
return flow;
}
int dinic(){
int ret=0;
while(spfa()){
memcpy(cir,head,sizeof(head));
ret+=dfs(ss,inf);
}
return ret;
}
int main()
{
memset(head,-1,sizeof(head));
scanf("%d",&N);
for(int i=1;i<=N;i++) scanf("%d",&data[i]);
scanf("%d%d%d%d%d",&p,&m,&f,&n,&s);
ss=0,tt=N*2+1;
add(ss,1,inf,p),add(1,ss,0,-p);
for(int i=1;i<=N;i++){
if(i!=N) add(i,i+1,inf,0),add(i+1,i,0,0);
add(i,tt,data[i],0),add(tt,i,0,0);
add(ss,N+i,data[i],0),add(N+i,ss,0,0);
if(m+i<=N) add(N+i,i+m,inf,f),add(m+i,i+N,0,-f);
if(n+i<=N) add(N+i,i+n,inf,s),add(n+i,i+N,0,-s);
}
dinic();
printf("%lld",ans);
}