Codeforces Round #765 (Div. 2) C. Road Optimization

题目链接：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;
}