洛谷 P2517 [HAOI2010]订货

题面

题意

给出每一个月的商品需求量和每一个月的生产成本以及仓库容量和贮存费用(每月,与贮存货物的量无关),问最小成本.

分析

一开始没有想到用费用流去做,且难以建图,但将名词转化一下即可
将货物的量当做流量,成本当做费用,先建一个超级源点(可以理解为工厂),连向每一个月的节点,因为生产数量无限制,故流量为INF,费用为成本,再建一个超级汇点(理解为客户),每一个月的节点连向它,流量为需求量,费用为0,再将相连两月连接,流量为仓库流量,费用为贮存费,跑一遍费用流即可.

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f3f
#define N 110
#define M 1010
using namespace std;

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

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

int dfs(int now,int mn)
{
    if(now==s)
        return mn;
    int 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 void bfs()
{
    int p,q;
    for(;!que.empty();que.pop());
    memset(d,INF,sizeof(d));
    memset(in,0,sizeof(in));
    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;
    ans+=d[t]*dfs(t,INF);
    bfs();
}

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