题目链接
另一种解法,不一样的思路。
点复杂度: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;
}