Heidi and Library (hard) 【CodeForces - 802C】【最大流最小费用】

题目链接

  另一种解法，不一样的思路。

---

点复杂度：O(N)

这次的解法，在点复杂度上做到了O(N)，可以说是一次点复杂度上的优化了，当然时间跟上一个差距不大。

  这次我的做法是把每一天的所需都加进去，也就是使得每一天的所用的书本都是新买的。但是这样就会很浪费的，我们有最大的可以减去的量，那么我们不妨去求最大的可以减去的量不就可以了吗。

  既然有了思路，我们就可以开始了，还是同样的限流，我们让最大的每条边的流量上限为（K-1），这次的建边是从（S -> 1 -> …… -> N -> T）这样的过程，那么接下去，我们考虑建边，按照上个解法的思路，我们如果要保存下来的话，其中有一段就会没有少去一本书了，就是这段时间内，书架上只能省去一个空位子给其中的一本书了，所以，我们将边建立为（pos[a[i]] -> i-1），保证了，我们从上一次的位置开始，一直到目前这一刻的前面一刻，一直给这本书留出一个位置来，所以权值的费用的代价为-c[a[i]]。

  总结：

从S -> 1 、所有的 i -> i+1 、N -> T都建立流为K-1，费用为0的边，表示书架上如果存了这本书，最多也就留K-1本其他的书了。

对所有的能找到上一个出现的位置的点，建立这样的边：pos[a[i]] -> i-1的流为1，费用为-c[a[i]]的边，表示着我们的这段区间是可以利用上一次的书的。

---

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define efs 1e-8
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 205, maxE = 1e3 + 7, S = 0;
int N, K, a[maxN], c[maxN], pos[maxN], head[maxN], cnt, T, sum;
struct Eddge
{
    int nex, u, v, flow, cost;
    Eddge(int a=-1, int b=0, int c=0, int d=0, int f=0):nex(a), u(b), v(c), flow(d), cost(f) {}
}edge[maxE];
inline void addEddge(int u, int v, int flow, int cost)
{
    edge[cnt] = Eddge(head[u], u, v, flow, cost);
    head[u] = cnt++;
}
inline void _add(int u, int v, int flow, int cost) { addEddge(u, v, flow, cost); addEddge(v, u, 0, -cost); }
int pre[maxN], dist[maxN], Flow[maxN];
bool inque[maxN];
queue<int> Q;
bool spfa()
{
    memset(pre, -1, sizeof(pre));   memset(dist, INF, sizeof(dist));    memset(inque, false, sizeof(inque));
    while(!Q.empty()) Q.pop();  Q.push(S);  inque[S] = true;    dist[S] = 0;    Flow[S] = INF;
    while(!Q.empty())
    {
        int u = Q.front();  Q.pop();    inque[u] = false;
        for(int i=head[u], v, f, c; ~i; i=edge[i].nex)
        {
            v = edge[i].v;  f = edge[i].flow;   c = edge[i].cost;
            if(f && dist[v] > dist[u] + c)
            {
                dist[v] = dist[u] + c;
                Flow[v] = min(Flow[u], f);
                pre[v] = i;
                if(!inque[v])
                {
                    inque[v] = true;
                    Q.push(v);
                }
            }
        }
    }
    return pre[T] ^ -1;
}
int EK()
{
    int ans = 0;
    while(spfa())
    {
        int now = T, las = pre[now];
        while(now)
        {
            edge[las].flow -= Flow[T];
            edge[las ^ 1].flow += Flow[T];
            now = edge[las].u;
            las = pre[now];
        }
        ans += dist[T] * Flow[T];
    }
    return ans;
}
inline void init()
{
    cnt = sum = 0;    T = N + 1;
    memset(head, -1, sizeof(head));
    memset(pos, 0, sizeof(pos));
}
int main()
{
    scanf("%d%d", &N, &K);
    init();
    for(int i=1; i<=N; i++) scanf("%d", &a[i]);
    for(int i=1; i<=N; i++) scanf("%d", &c[i]);
    for(int i=1; i<=N; i++) sum += c[a[i]];
    for(int i=1; i<N; i++) _add(i, i + 1, K - 1, 0);
    _add(S, 1, K - 1, 0);
    pos[a[1]] = 1;
    for(int i=2; i<=N; i++)
    {
        if(pos[a[i]])
        {
            if(pos[a[i]] + 1 == i) sum -= c[a[i]];
            else _add(pos[a[i]], i - 1, 1, -c[a[i]]);
        }
        pos[a[i]] = i;
    }
    _add(N, T, K - 1, 0);
    printf("%d\n", sum + EK());
    return 0;
}