Submission #850973

#	Time	Username	Problem	Language	Result	Execution time	Memory
850973		NamPE	Stove (JOI18_stove)	C++17	0 / 100	180 ms	262144 KiB

stove

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

const ll MAXN = 5e3;

vector<vector<vector<ll>>> dp(MAXN + 1, vector<vector<ll>>(MAXN + 1, vector<ll>(2, -1)));

ll dfs(vector<ll>& a, ll i, ll k, ll state) {
    if (i == a.size()) {
        return (!state ? 0 : INT_MAX);
    }

    ll res = INT_MAX;

    if (!state && k) {
        res = min(res, dfs(a, i + 1, k - 1, 0) + 1);
        res = min(res, dfs(a, i + 1, k - 1, 1) - a[i]);
    } else if (state) {
        res = min(res, dfs(a, i + 1, k, 1));
        res = min(res, dfs(a, i + 1, k, 0) + a[i] + 1);
    }

    return res;
}

int main() {
    ios::sync_with_stdio(0);
    cout.tie(0);
    cin.tie(0);

    ll n, k;
    cin >> n >> k;

    vector<ll> a(n);

    for (ll& i : a) {
        cin >> i;
    }

    cout << dfs(a, 0, k, 0);
}

Compilation message (stderr)

stove.cpp: In function 'll dfs(std::vector<long long int>&, ll, ll, ll)':
stove.cpp:11:11: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   11 |     if (i == a.size()) {
      |         ~~^~~~~~~~~~~

• Subtask 1: 0 / 20
• Subtask 2: 0 / 30
• Subtask 3: 0 / 50