题目链接:https://codeforces.com/contest/1625/problem/C
一开始思路是想dp[i][j],去除的站的数目为i个,终点站为第j个站(站从0开始数,令终点站为第n个),0站和n站是不能被去除的。
然后发现不知道dp[i-1][j-1]最后去除的站是哪一个就算不出dp[i][j],然后变成dp[i][j][k],k表示最后去除的站。
然后又发现之前去除的站是有关联的,如果最后去除的站为k站,k-1站是什么状态就不知道了。那就把k的含义改成左边第一个没有去除的站。调试一会儿答案就有了。
但是最后由于内存占用过大不行,以为用滚动数组,到结束都没调处来。结束完十来分钟发现要用到的只有dp[i]和dp[i-1]两个二维数组,那么可以将数组变为dp[2][N][N],轮流用0和1表示i和i-1即可。
#include <bits/stdc++.h>
using namespace std;
const int N = 5e2 + 5;
const int INF = 0x3f3f3f3f;
int t, n, l, k, d[N], a[N], dp[2][N][N];
// dp[i][j][k] 终点为第j个站, 去除i个站,k表示左边第一个没去除的站的位置, 不包括当前位置
int main(void) {
// freopen("in.txt", "r", stdin);
ios::sync_with_stdio(false);
cin >> n >> l >> k;
for (int i = 0; i < n; i++) cin >> d[i];
for (int i = 0; i < n; i++) cin >> a[i];
memset(dp, INF, sizeof(dp));
d[n] = l;
dp[0][1][0] = a[0] * d[1];
for (int i = 2; i <= n; i++)
dp[0][i][i - 1] = dp[0][i - 1][i - 2] + a[i - 1] * (d[i] - d[i - 1]);
int ans = dp[0][n][n - 1];
for (int i = 1; i <= k; i++) {
int p = i & 1;
for (int j = 2; j <= n; j++) {
for (int x = j - 2; x >= 0; x--) {
dp[p][j][x] = dp[p ^ 1][j - 1][x] + a[x] * (d[j] - d[j - 1]);
dp[p][j][j - 1] = min(dp[p][j][j - 1], dp[p][j - 1][x]);
}
dp[p][j][j - 1] += a[j - 1] * (d[j] - d[j - 1]);
}
for (int x = n - 1; x >= 0; x--)
ans = min(ans, dp[p][n][x]);
}
cout << ans << "\n";
return 0;
}