餐巾计划问题 费用流

                                                                                                                                                餐巾计划问题
                                                      时间限制: 1 s                                                            空间限制: 128000 KB                                                               题目等级 : 钻石 Diamond
 查看运行结果
题目描述 Description
一个餐厅在相继的 N 天里，每天需用的餐巾数不尽相同。假设第 i 天需要 ri块餐巾(i=1，2，…，N)。餐厅可以购买新的餐巾，每块餐巾的费用为 p 分；或者把旧餐巾送到快洗部，洗一块需 m 天，其费用为 f 分；或者送到慢洗部，洗一块需 n 天(n>m)，其费用为 s<f 分。
每天结束时，餐厅必须决定将多少块脏的餐巾送到快洗部，多少块餐巾送到慢洗部，以及多少块保存起来延期送洗。但是每天洗好的餐巾和购买的新餐巾数之和，要满足当天的需求量。
试设计一个算法为餐厅合理地安排好 N 天中餐巾使用计划，使总的花费最小。 
编程找出一个最佳餐巾使用计划.


输入描述 Input Description
第 1 行有 6 个正整数 N,p,m,f,n,s。N 是要安排餐巾使用计划的天数；p 是每块新餐巾的费用；m 是快洗部洗一块餐巾需用天数；f 是快洗部洗一块餐巾需要的费用；n 是慢洗部洗一块餐巾需用天数；s 是慢洗部洗一块餐巾需要的费用。接下来的 N 行是餐厅在相继的 N 天里，每天需用的餐巾数。


输出描述 Output Description
将餐厅在相继的 N 天里使用餐巾的最小总花费输出


样例输入 Sample Input
3 10 2 3 3 2


5


6


7


样例输出 Sample Output
145

数据范围及提示 Data Size & Hint

分析：题意有点不明啊，一块毛巾洗m天，这里是一队毛巾洗m天。。。

建图：

1、S向每一天i建边add(S,i,INF,0);

2、每一天向终点T建边add(i,T,a[i],0);

3、S向每一个洗衣（天）建边add(S,i+N,a[i],0);

4、慢洗：每一次洗衣向可以使用的天建边add(i+N,i+m,INF,f);

5、快洗：每一次洗衣向可以使用的天建边add(i+N,i+n,INF,s);

6、每一次洗衣向下一次洗衣建边add(i+N,i+N+1,INF,0);

代码：

//如果要求最大费用的话 只需在加边的时候加-的边  输出时输出-ans即可
#pragma comment(linker,"/STACK:102400000,102400000")
#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <string>
#include <math.h>
#include <queue>
#include <stack>
#include <map>
#include <set>
using namespace std;
typedef long long ll;   //记得必要的时候改成无符号
const int maxn=1600;
const int maxm=1000005;
const int INF=1000000000;
struct EdgeNode
{
    int from;
    int to;
    int flow;
    int cost;
    int next;
}edge[maxm];
int head[maxn],cnt;
void add(int x,int y,int z,int c)
{
    edge[cnt].from=x;edge[cnt].to=y;edge[cnt].flow=z;edge[cnt].cost=c;edge[cnt].next=head[x];head[x]=cnt++;
    edge[cnt].from=y;edge[cnt].to=x;edge[cnt].flow=0;edge[cnt].cost=-c;edge[cnt].next=head[y];head[y]=cnt++;
    //printf("%d %d %d %d\n",x,y,z,c);
}

void init()
{
    cnt=0;
    memset(head,-1,sizeof(head));
}

int S,T,n,m;
int d[maxn],in[maxn],pre[maxn];
queue<int>Q;
bool spfa(int S,int T)
{
    int u,v,f,c;
    while(!Q.empty())Q.pop();
    memset(in,0,sizeof(in));
    for(int i=0;i<=n;i++)d[i]=INF;
    d[S]=0;
    Q.push(S);
    while(!Q.empty())
    {
        u=Q.front(); Q.pop(); in[u]=0;
        for(int i=head[u];i!=-1;i=edge[i].next){
            v=edge[i].to; f=edge[i].flow; c=edge[i].cost;
            if(f&&d[u]+c<d[v]){
                d[v]=d[u]+c; pre[v]=i;
                if(!in[v]){
                    in[v]=1;
                    Q.push(v);
                }
            }
        }
    }
    if(d[T]==INF)return false;
    return true;
}

int MCMF(int S,int T,int need=0)
{
    int u;
    int max_flow=0;
    int min_cost=0;
    while(spfa(S,T))
    {
        int flow=INF;
        u=T;
        while(u!=S){
            flow=min(flow,edge[pre[u]].flow);
            u=edge[pre[u]].from;
        }
        u=T; max_flow+=flow; min_cost+=d[T]*flow;
        while(u!=S){
            edge[pre[u]].flow-=flow;
            edge[pre[u]^1].flow+=flow;
            u=edge[pre[u]].from;
        }
    }
    if(max_flow<need) return -1;
    return min_cost;
}

int a[maxn];
int main()
{
    int i,j,N,p,f,s,k,ans;
    while(~scanf("%d%d%d%d%d%d",&N,&p,&m,&f,&n,&s))
    {
        init(); S=0; T=2*N;
        for(i=1;i<=N;i++)scanf("%d",&a[i]);
        for(i=1;i<=N;i++){
            add(S,i,INF,p);
            add(i,T,a[i],0);
        }
        for(i=1;i<N;i++){
            add(S,i+N,a[i],0);
            if(i+m<=N)add(i+N,i+m,INF,f);
            if(i+n<=N)add(i+N,i+n,INF,s);
        }
        for(i=1;i<N-1;i++)add(i+N,i+1+N,INF,0);
        n=T;
        printf("%d\n",MCMF(S,T));
    }
    return 0;
}