Submission #1331528

#	Time	Username	Problem	Language	Result	Execution time	Memory
1331528		kawhiet	Safety (NOI18_safety)	C++20	7 / 100	2089 ms	327680 KiB

safety

#include <bits/stdc++.h>
using namespace std;

#define int long long

constexpr int inf = 1e18;

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, h;
    cin >> n >> h;
    vector<int> a(n);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
    }
    int m = ranges::max(a);
    vector<map<int, int>> dp(n);
    // vector<vector<int>> dp(n, vector<int>(m + 1, inf));
    for (int i = 0; i <= m; i++) {
        dp[0][i] = abs(i - a[0]);
    }
    for (int i = 1; i < n; i++) {
        for (int j = 0; j <= m; j++) {
            int l = max(0LL, j - h);
            int r = min(m, j + h);
            for (int k = l; k <= r; k++) {
                if (!dp[i].count(j)) {
                    dp[i][j] = inf;
                }
                dp[i][j] = min(dp[i][j], dp[i - 1][k] + abs(a[i] - j));
            }
        }
    }
    int ans = inf;
    for (auto [x, y] : dp[n - 1]) {
        ans = min(ans, y);
    }
    cout << ans << '\n';
    return 0;
}

• Subtask 1: 3 / 3
• Subtask 2: 4 / 4
• Subtask 3: 0 / 9
• Subtask 4: 0 / 5
• Subtask 5: 0 / 6
• Subtask 6: 0 / 11
• Subtask 7: 0 / 11
• Subtask 8: 0 / 22
• Subtask 9: 0 / 29