餐巾计划问题（费用流）

P1251 餐巾计划问题

题目描述

一个餐厅在相继的 $N$ 天里,每天需用的餐巾数不尽相同。假设第 $i$ 天需要$r_i$ 块餐巾( $i=1,2,...,N$)。餐厅可以购买新的餐巾,每块餐巾的费用为$p$分;或者把旧餐巾送到快洗部,洗一块需 $m$ 天,其费用为 $f$ 分;或者送到慢洗部,洗一块需 $n$ 天( $n>m$ ),其费用为 $s$ 分( $s<f$ )。

每天结束时,餐厅必须决定将多少块脏的餐巾送到快洗部,多少块餐巾送到慢洗部,以及多少块保存起来延期送洗。但是每天洗好的餐巾和购买的新餐巾数之和,要满足当天的需求量。

试设计一个算法为餐厅合理地安排好 $N$ 天中餐巾使用计划,使总的花费最小。编程找出一个最佳餐巾使用计划。

输入格式：

由标准输入提供输入数据。文件第 1 行有 1 个正整数 $N$ ，代表要安排餐巾使用计划的天数。

接下来的 $N$ 行是餐厅在相继的 $N$ 天里,每天需用的餐巾数。

最后一行包含5个正整数 $p,m,f,n,s$ 。 $p$ 是每块新餐巾的费用; $m$ 是快洗部洗一块餐巾需用天数; $f$ 是快洗部洗一块餐巾需要的费用; $n$ 是慢洗部洗一块餐巾需用天数; $s$ 是慢洗部洗一块餐巾需要的费用。

输出格式：

将餐厅在相继的 $N$ 天里使用餐巾的最小总花费输出

输入样例#1：

3
1 7 5 
11 2 2 3 1

输出样例#1：

134

说明

$N≤2000$
$r_i≤10000000$
$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);
}