Acwing 2184. 餐巾计划问题

餐巾计划问题

思路

将一天的毛巾 分为两部分 一部分为旧毛巾 一部分为新毛巾即所需要的毛巾
对于新毛巾所被连的边 1-新买的毛巾 2-快洗店送来的毛巾 3-慢洗店送来的毛巾
所连的边 到汇点T即完成任务的边
对于老毛巾所被连的边 1-一天用完的毛巾 2-前一天剩下的所有旧毛巾
所连的边 1-保留到第二天的毛巾 2-送到快洗店的毛巾 3-送到慢洗店的毛巾

注意 要用S连向旧毛巾来输送旧毛巾的流 因为当天用完的新毛巾需要连向汇点T去完成任务 如果用新毛巾直接连向旧毛巾来输送旧毛巾的流 会导致当天旧毛巾可能无流 因为都流向了T

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#include <cstdio>

using namespace std;

const int N = 2010, M = 20010, INF = 1e8;

int n, p, x, xp, y, yp, S, T;
int h[N], e[M], ne[M], f[M], w[M], idx;
int d[N], pre[N], incf[N];
bool st[N];

queue<int> q;

void add(int a, int b, int c, int d)
{
    e[idx] = b, f[idx] = c, w[idx] = d, ne[idx] = h[a], h[a] = idx ++ ;
    e[idx] = a, f[idx] = 0, w[idx] = -d, ne[idx] = h[b], h[b] = idx ++ ;
}

bool spfa()
{
    memset(d, 0x3f, sizeof d);
    memset(incf, 0, sizeof incf);
    q.push(S), d[S] = 0, incf[S] = INF;
    while (q.size())
    {
        int t = q.front();
        q.pop();
        st[t] = false;
        for (int i = h[t]; ~i; i = ne[i])
        {
            int ver = e[i];
            if (f[i] && d[ver] > d[t] + w[i])
            {
                d[ver] = d[t] + w[i];
                pre[ver] = i;
                incf[ver] = min(f[i], incf[t]);
                if (!st[ver])
                {
                    q.push(ver);
                    st[ver] = true;
                }
            }
        }
    }
    return incf[T] > 0;
}

int EK()
{
    int cost = 0;
    while (spfa())
    {
        int t = incf[T];
        cost += t * d[T];
        for (int i = T; i != S; i = e[pre[i] ^ 1])
        {
            f[pre[i]] -= t;
            f[pre[i] ^ 1] += t;
        }
    }
    return cost;
}



int main()
{
    scanf("%d%d%d%d%d%d", &n, &p, &x, &xp, &y, &yp);
    
    memset(h, -1, sizeof h);
    S = 0, T = n * 2 + 1;
    
    for (int i = 1; i <= n; i ++ )
    {
        int w;
        scanf("%d", &w);
        add(S, i + n, INF, p);
        add(i + n, T, w, 0);
        add(S, i, w, 0);
        if (i < n) add(i, i + 1, INF, 0);
        if (i + x <= n) add(i, i + x + n, INF, xp);
        if (i + y <= n) add(i, i + y + n, INF, yp);
    }
    
    printf("%d\n", EK());
    
    return 0;
}