洛谷 P1251 餐巾计划问题

题面

题意

有一家餐馆,每天需要一些餐巾,用完之后都会变脏,有两个洗餐巾的地方,给出单价和需要天数(可以放入任意数量的餐巾),每天可以买一些餐巾,问最小费用.

方法

这题建图主要难点是汇点无法确定,因为餐巾可以通过清洗再次使用,而不能像流量一样直接流向汇点.
解决方法是每天拆成两个节点,一个A点用于处理干净的餐巾,一个B点用于处理脏的餐巾,干净餐巾经过使用后变脏可以理解为干净餐巾从A流向超级汇点,同时又有同样流量的干净餐巾从超级源点流向B点,接着B点再根据两个清洗部的速度与价格建边流向后几天的A点,另外需要注意的是,餐馆可以囤积干净餐巾,因而相邻两天的A点要建一条流量无限大,费用为0的边.

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f3f3f3f3f3f
#define ll long long
#define N 4100
using namespace std;

ll n,m,s,t,ans,bb=1,first[N],d[N],last[N],B[N];
bool in[N];
struct Bn
{
    ll to,next,cst,quan;
}bn[N*20];
queue<ll>que;

inline void add(ll u,ll v,ll w,ll z)
{
    bb++;
    bn[bb].to=v;
    bn[bb].cst=z;
    bn[bb].quan=w;
    bn[bb].next=first[u];
    first[u]=bb;
}

ll dfs(ll now,ll mn)
{
    if(now==s) return mn;
    ll res;
    res=dfs(last[now],min(bn[B[now]].quan,mn));
    bn[B[now]].quan-=res;
    bn[B[now]^1].quan+=res;
    return res;
}

inline bool bfs()
{
    ll p,q;
    memset(d,0x3f,sizeof(d));
    que.push(s);
    d[s]=0;
    for(;!que.empty();)
    {
        q=que.front();
        que.pop();
        in[q]=0;
        for(p=first[q];p!=-1;p=bn[p].next)
        {
            if(d[bn[p].to]<=d[q]+bn[p].cst||!bn[p].quan) continue;
            d[bn[p].to]=d[q]+bn[p].cst;
            last[bn[p].to]=q;
            B[bn[p].to]=p;
            if(!in[bn[p].to])
            {
                in[bn[p].to]=1;
                que.push(bn[p].to);
            }
        }
    }
    if(d[t]==INF) return 0;
    ans+=dfs(t,INF)*d[t];
    return 1;
}

int main()
{
    memset(first,-1,sizeof(first));
    ll i,j,p,q;
    cin>>n;
    t=n*2+1;
    for(i=1;i<=n;i++)
    {
        scanf("%lld",&p);
        add(s,i+n,p,0);
        add(i+n,s,0,0);
        add(i,t,p,0);
        add(t,i,0,0);
        if(i!=n)
        {
            add(i,i+1,INF,0);
            add(i+1,i,0,0);
        }
    }
    cin>>p;
    for(i=1;i<=n;i++)
    {
        add(s,i,INF,p);
        add(i,s,0,-p);
    }
    cin>>p>>q;
    for(i=1;i+p<=n;i++)
    {
        add(i+n,i+p,INF,q);
        add(i+p,i+n,0,-q);
    }
    cin>>p>>q;
    for(i=1;i+p<=n;i++)
    {
        add(i+n,i+p,INF,q);
        add(i+p,i+n,0,-q);
    }
    for(;bfs(););
    cout<<ans;
}