Codeforces Round #765 C. Road Optimization

C. Road Optimization

原题链接

题目大意：总长为 $l$ 的路，有 $n$ 个限速牌，限速的方式主要看下图案例（它是 分钟/每米）：

现在给出每个限速牌的位置 $d_i$ 和限速 $a_i$ ，问最多拆除 $k$ 个限速牌的情况下（第一个限速牌不能拆），走完这段路的最短时间。

看数据范围和时间限制，可以想到是个 $n^3$ 的写法，应该是个 $DP$ 了。

我们定义 $dp[i][0/1][j]$ 的含义为：走完从起点到 $d[i+1]$ 这段路，第 $i$ 个限速牌是否被拆除（这里 $1$ 为保留， $0$ 为拆除），此时一共已经拆除 $j$ 个限速牌的最短时间。如果 $i$ 是最后一个限速牌，则 $d[i+1]=l$ 。

因为第一个点比较特殊，我们先初始化 $i=0$ 的情况，对于第一个点而言，只有 $dp[0][1][0]$ 有值，其值为 $a[0]\ast (p[1]-p[0])$ ，其它都应该赋值为正无穷。

然后我们可以分两种情况讨论：

① $dp[i][1][j]$ ：$dp[i][1][j]=min(dp[i-1][1][j],dp[i-1][0][j])+a[i]\ast (p[i+1]-p[i])$
也就是取上一步的最大值，加上最后没算的一段，因为 $i$ 没拆，所以速度自然是 $a_i$

② $dp[i][0][j]$ 的情况的话，那么重点是知道前一个没拆的是什么，所以这里要遍历所有前面的路标 $l(l<i)$ ，找到最小的 $dp[l][1][j-(i-l)]+a[l]\ast (p[i+1]-p[l+1])$

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 5e2 + 2;

int n, l, k;

ll dp[N][2][N], p[N], a[N];

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    scanf("%d%d%d", &n, &l, &k);
    for (int i = 0; i < n; ++i) {
        scanf("%lld", &p[i]);
    }
    p[n] = l;
    for (int i = 0; i < n; ++i) {
        scanf("%lld", &a[i]);
    }
    dp[0][1][0] = a[0] * (p[1] - p[0]);
    for (int i = 1; i <= k; ++i) {
        dp[0][1][i] = 1e16;
    }
    for (int i = 0; i <= k; ++i) {
        dp[0][0][i] = 1e16;
    }
    for (int i = 1; i < n; ++i) {
        for (int j = 0; j <= k; ++j) {
            dp[i][1][j] = min(dp[i - 1][1][j], dp[i - 1][0][j]) + a[i] * (p[i + 1] - p[i]);
            dp[i][0][j] = 1e16;
            for (int l = 0; l < i; ++l) {
                if (j >= (i - l)) dp[i][0][j] = min(dp[i][0][j], dp[l][1][j - (i - l)] + a[l] * (p[i + 1] - p[l + 1]));
            }
        }
    }
    ll ans = 1e16;
    for (int i = 0; i <= k; ++i) {
        ans = min(ans, min(dp[n - 1][0][i], dp[n - 1][1][i]));
    }
    printf("%lld", ans);
}