bzoj 2424: [HAOI2010]订货 费用流

题意

某公司估计市场在第i个月对某产品的需求量为Ui，已知在第i月该产品的订货单价为di，上个月月底未销完的单位产品要付存贮费用m，假定第一月月初的库存量为零，第n月月底的库存量也为零，问如何安排这n个月订购计划，才能使成本最低？每月月初订购，订购后产品立即到货，进库并供应市场，于当月被售掉则不必付存贮费。假设仓库容量为S。
n<=50

分析

看的第一眼以为是dp，发现好像未必满足单调性，优化不懂。查了一下发现可以费用流。一开始以为进货后要先存到仓库，这样的话就要拆点，后来发现只要进货就可以直接卖掉，那样的话就可以直接做了。
总而言之就是一句话：傻逼费用流。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int N=55;
const int inf=0x3f3f3f3f;

int n,m,S,cnt,last[N],s,t,dis[N],vis[N],pre[N],ans;
struct edge{int from,to,c,w,next;}e[N*N*2];
queue<int> que;

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void addedge(int u,int v,int c,int w)
{
    e[++cnt].from=u;e[cnt].to=v;e[cnt].c=c;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].from=v;e[cnt].to=u;e[cnt].c=0;e[cnt].w=-w;e[cnt].next=last[v];last[v]=cnt;
}

bool spfa()
{
    for (int i=s;i<=t;i++) dis[i]=inf;
    dis[s]=0;que.push(s);vis[s]=1;
    while (!que.empty())
    {
        int u=que.front();que.pop();
        for (int i=last[u];i;i=e[i].next)
            if (e[i].c&&dis[u]+e[i].w<dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+e[i].w;
                pre[e[i].to]=i;
                if (!vis[e[i].to]) vis[e[i].to]=1,que.push(e[i].to);
            }
        vis[u]=0;
    }
    if (dis[t]<inf) return 1;
    else return 0;
}

void mcf()
{
    int mn=inf,x=t;
    while (pre[x])
    {
        mn=min(mn,e[pre[x]].c);
        x=e[pre[x]].from;
    }
    ans+=dis[t]*mn;x=t;
    while (pre[x])
    {
        e[pre[x]].c-=mn;
        e[pre[x]^1].c+=mn;
        x=e[pre[x]].from;
    }
}

int main()
{
    n=read();m=read();S=read();
    s=0;t=n+1;cnt=1;
    for (int i=1;i<=n;i++)
    {
        int x=read();
        addedge(i,t,x,0);
    }
    for (int i=1;i<=n;i++)
    {
        int x=read();
        addedge(s,i,inf,x);
    }
    for (int i=1;i<n;i++) addedge(i,i+1,S,m);
    while (spfa()) mcf();
    printf("%d",ans);
    return 0;
}