cogs 461. [网络流24题] 餐巾

1. [网络流24题] 餐巾
   ★★★ 输入文件：napkin.in 输出文件：napkin.out 简单对比
   时间限制：5 s 内存限制：128 MB
   【问题描述】

   一个餐厅在相继的N天里，第i天需要Ri块餐巾(i=l，2，…，N)。餐厅可以从三种途径获得餐巾。
   (1)购买新的餐巾，每块需p分；
   (2)把用过的餐巾送到快洗部，洗一块需m天，费用需f分(f < p)。如m=l时，第一天送到快洗部的餐巾第二天就可以使用了，送慢洗的情况也如此。
   (3)把餐巾送到慢洗部，洗一块需n天(n>m)，费用需s分(s < f)。
   在每天结束时，餐厅必须决定多少块用过的餐巾送到快洗部，多少块送慢洗部。在每天开始时，餐厅必须决定是否购买新餐巾及多少，使洗好的和新购的餐巾之和满足当天的需求量Ri，并使N天总的费用最小。

【输入】
输入文件共 3 行，第 1 行为总天数；第 2 行为每天所需的餐巾块数；第 3 行为每块餐巾的新购费用 p ，快洗所需天数 m ，快洗所需费用 f ，慢洗所需天数 n ，慢洗所需费用 s 。
【输出】
一行，最小的费用
【样例】
napkin.in
3
3 2 4
10 1 6 2 3
napkin.out
64
【数据规模】
n<=200,Ri<=50

---

【分析】
神奇二分建图法。

---

【代码】

//461. [网络流24题] 餐巾
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define inf 1e9+7
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=100005<<2;
queue <int> q;
int N,n,m,s,t,F,S,P,cnt,ans;
bool vis[mxn];
int head[mxn],dis[mxn],pre[mxn];
struct node {int from,to,next,d,flow;} f[mxn];
inline void add(int u,int v,int d,int flow)
{
    f[++cnt].to=v,f[cnt].from=u,f[cnt].next=head[u],f[cnt].d=d,f[cnt].flow=flow,head[u]=cnt;
    f[++cnt].to=u,f[cnt].from=v,f[cnt].next=head[v],f[cnt].d=-d,f[cnt].flow=0,head[v]=cnt;
}
inline void spfa()
{
    int i,u,v,d,flow;
    memset(dis,0x3f,sizeof dis);
    memset(pre,-1,sizeof pre);
    M(vis);
    q.push(s);
    dis[s]=0;
    vis[s]=1;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        vis[u]=0;
        for(i=head[u];i;i=f[i].next)
        {
            v=f[i].to,d=f[i].d,flow=f[i].flow;
            if(dis[v]>dis[u]+d && flow>0)
            {
                dis[v]=dis[u]+d;
                pre[v]=i;   //记录前驱 
                if(!vis[v]) vis[v]=1,q.push(v);
            }
        }
    }
}
inline void maxflow()
{
    int i,u,v,d;
    spfa();
    while(pre[t]!=-1)
    {
        int tmp=inf;
        for(i=pre[t];i!=-1;i=pre[f[i].from])
          tmp=min(tmp,f[i].flow);
        ans+=dis[t]*tmp;
        for(i=pre[t];i!=-1;i=pre[f[i].from])
        {
            f[i].flow-=tmp;
            if(i&1) f[i+1].flow+=tmp;
            else f[i-1].flow+=tmp;
        }
        spfa();
    }
}
int main()
{
    freopen("napkin.in","r",stdin);
    freopen("napkin.out","w",stdout);
    int i,j,k,u,v,x;
    scanf("%d",&N);
    s=0,t=N+N+1;
    fo(i,1,N) 
    {
        scanf("%d",&x);
        add(s,i,0,x),add(i+N,t,0,x);
    }
    scanf("%d%d%d%d%d",&P,&m,&F,&n,&S);
    fo(i,1,N-1) add(i,i+1,0,inf);
    fo(i,N+1,N+N) add(s,i,P,inf);
    fo(i,1,N) if(i+N+m<=N+N) add(i,i+N+m,F,inf);
    fo(i,1,N) if(i+N+n<=N+N) add(i,i+N+n,S,inf);
    maxflow();
    printf("%d\n",ans);
    return 0;
}