BZOJ1221: [HNOI2001] 软件开发

题目链接

【分析】

和餐巾计划基本相同。

【建图】

拆点，每个点x拆为x’和x”。

1. S向每个点x’建边，容量为n[i]，费用为0。表示当天用了n[i]。
2. 每个x”向T建边，容量为n[i]，费用为0。表示当天需要n[i]。
3. S向每个点x’’建边，容量为INF，费用为f。表示当天通过购买来获得毛巾。
4. 每个点x’向(x+1)’建边，容量为INF，费用为0。表示当天用完的毛巾留到下一天处理。
5. 若x+a+1<=n，建边x’–>(x+a+1)”，容量为INF，费用为fa。表示当天用完的毛巾送去第a种消毒方式进行消毒。注意：当天用完后第二天开始消毒，消毒a天，所以第x+a+1天才可以使用。下同。
6. 若x+b+1<=n，建边x’–>(x+b+1)”，容量为INF，费用为fb。

【建模分析】

若还不太理解，引用别人写的餐巾计划的建模分析（餐巾计划中消毒后第x+m天就可用，下面已修改为本题的要求）：

这个问题的主要约束条件是每天的餐巾够用，而餐巾的来源可能是最新购买，也可能是前几天送洗，今天刚刚洗好的餐巾。每天用完的餐巾可以选择送到快洗部或慢洗部，或者留到下一天再处理。
经过分析可以把每天要用的和用完的分离开处理，建模后就是二分图。二分图X集合中顶点Xi表示第i天用完的餐巾，其数量为ri，所以从S向Xi连接容量为ri的边作为限制。Y集合中每个点Yi则是第i天需要的餐巾，数量为ri，与T连接的边容量作为限制。每天用完的餐巾可以选择留到下一天（Xi->Xi+1），不需要花费，送到快洗部(Xi->Yi+m+1)，费用为f，送到慢洗部(Xi->Yi+n+1)，费用为s。每天需要的餐巾除了刚刚洗好的餐巾，还可能是新购买的(S->Yi)，费用为p。
在网络上求出的最小费用最大流，满足了问题的约束条件（因为在这个图上最大流一定可以使与T连接的边全部满流，其他边只要有可行流就满足条件），而且还可以保证总费用最小，就是我们的优化目标。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 2005
#define M 12005
#define INF 1000000000
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,K,F,Fa,Fb,cnt=1,S,T,ans;
int b[M],p[N],nextedge[M],w[M],c[M],cur[N];
int Dis[N],Pre[N];
bool Flag[N];

void Add(int x,int y,int z,int cost)
{
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
    w[cnt]=z;
    c[cnt]=cost;
}

void Anode(int x,int y,int z,int cost){
    Add(x,y,z,cost);Add(y,x,0,-cost);
}

void Input_Init()
{
    n=read(),m=read(),K=read(),F=read(),Fa=read(),Fb=read();
    T=n<<1|1;
    for(int i=1;i<=n;i++)
    {
        static int x;
        x=read();
        Anode(0,i,x,0);
        Anode(i+n,T,x,0);
        Anode(0,i+n,INF,F);
        if(i<n) Anode(i,i+1,INF,0);
        if(i+m+1<=n) Anode(i,i+m+n+1,INF,Fa);
        if(i+K+1<=n) Anode(i,i+K+n+1,INF,Fb);
    }
}

bool Bfs()
{
    queue<int>q;
    q.push(S);
    for(int i=0;i<=T;i++) Dis[i]=INF;
    Dis[S]=0;
    while(!q.empty())
    {
        int k=q.front();q.pop();Flag[k]=0;
        for(int i=p[k];i;i=nextedge[i])
        {
            int v=b[i],f=w[i];
            if(Dis[v]>Dis[k]+c[i]&&f)
            {
                Dis[v]=Dis[k]+c[i];
                Pre[v]=i;
                if(!Flag[v])
                {
                    Flag[v]=1;
                    q.push(v);
                }
            }
        }
    }
    return Dis[T]!=INF;
}

void Mcf()
{
    int Maxf=INF;
    for(int i=Pre[T];i;i=Pre[b[i^1]])
        Maxf=min(Maxf,w[i]);
    for(int i=Pre[T];i;i=Pre[b[i^1]])
    {
        w[i]-=Maxf;w[i^1]+=Maxf;
        ans+=Maxf*c[i];
    }
}

void MCF()
{
    while(Bfs())
        Mcf();
    printf("%d\n",ans);
}

int main()
{
    Input_Init();
    MCF();
    return 0;
}