题意: 一个餐厅在接下来
n
n
n天里每天需要
r
i
r_i
ri个餐巾,可以购买餐巾,单价为
p
p
p,可以送到快洗店,花
m
m
m天和
f
f
f钱,送到慢洗店花
n
n
n天和
s
s
s钱。每天结束时,餐厅必须决定将多少块脏的餐巾送到快洗部,多少块餐巾送到慢洗部,以及多少块保存起来延期送洗。但是每天洗好的餐巾和购买的新餐巾数之和,要满足当天的需求量。
试设计一个算法为餐厅合理地安排好
N
N
N 天中餐巾使用计划,使总的花费最小。编程找出一个最佳餐巾使用计划。
思路: 每天开始的时候只有干净的餐巾可使用,每天结束的时候仅有脏的餐巾需要操作。于是将每天拆成两个点:起始点与结束点,分别处理不同时间段所需操作。
由于只有每天结束时才有餐巾操作,所以应该把结束点于源点连起来。
结束点有三种可以的操作:
- 将餐巾送到快洗店,将结束点 i i i与 i + m i+m i+m天的起始点连接,容量为 I N F INF INF,费用为 f f f。
- 将餐巾送到慢洗店,就是将结束点 i i i与 i + n i+n i+n天的起始点连接,容量为 I N F INF INF,费用为 s s s。
- 可能接下来几天需要餐巾很少,今天的餐巾可能就用不着了,不用洗了,不过为了好建边,可以把结束点 i i i与 i + 1 i+1 i+1的结束点连接,容量为 I N F INF INF,费用是0。
起始点只有一种操作:购买餐巾,从源点购买,源点与第
i
i
i天的起始点连接,容量为
I
N
F
INF
INF,费用为
p
p
p。
剩下的就是将源点与每个点的结束点连接,因为只有结束点才有脏餐巾操作。起始点连向他们的汇点,容量为这天所需的餐巾数量,费用为0。
#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N = 1e5 + 10;
int n, m, s, t, h[N], cnt, dis[N], inq[N], pre[N], last[N], flow[N];
struct node {
int v, w, c, nt;
} no[N];
void add(int u, int v, int w, int c) {
no[cnt] = node{v, w, c, h[u]};
h[u] = cnt++;
no[cnt] = node{u, 0, -c, h[v]};
h[v] = cnt++;
}
int spfa() {
memset(inq, 0, sizeof inq);
memset(dis, INF, sizeof dis);
dis[s] = 0, inq[s] = 1, flow[s] = INF;
queue<int> q;
q.push(s);
while(!q.empty()) {
int u = q.front();
q.pop(), inq[u] = 0;
for(int i = h[u]; ~i; i = no[i].nt) {
int v = no[i].v;
if(no[i].w > 0 && dis[v] > dis[u] + no[i].c) {
dis[v] = dis[u] + no[i].c;
pre[v] = u, last[v] = i;
flow[v] = min(flow[u], no[i].w);
if(!inq[v])
inq[v] = 1, q.push(v);
}
}
}
return dis[t] != INF;
}
ll mcmf() {
ll ans = 0;
while(spfa()) {
for(int i = t; i != s; i = pre[i]) {
no[last[i]].w -= flow[t];
no[last[i] ^ 1].w += flow[t];
}
ans += 1ll * flow[t] * dis[t];
}
return ans;
}
int p, a, b, f, r;
int main() {
memset(h, -1, sizeof h);
scanf("%d", &n);
t = n * 2 + 1;
for(int k, i = 1; i <= n; i++) {
scanf("%d", &k);
add(0, i, k, 0), add(i + n, t, k, 0);
}
scanf("%d%d%d%d%d", &p, &a, &f, &b, &r);
for (int i = 1; i <= n; i++) {
add(0, i + n, INF, p); //从源点购买餐巾
if (i + 1 <= n)
add(i, i + 1, INF, 0); //把今天的脏毛巾拖到明天
if (i + a <= n)
add(i, i + n + a, INF, f); //把餐巾送到快洗部
if (i + b <= n)
add(i, i + n + b, INF, r); //把餐巾送到慢洗部
}
printf("%lld\n", mcmf());
return 0;
}