Submission #850970

#	Time	Username	Problem	Language	Result	Execution time	Memory
850970		NamPE	Stove (JOI18_stove)	C++17	20 / 100	1050 ms	4440 KiB

stove

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

ll dp[(ll)(5e2 + 1)][(ll)(5e2 + 1)][2];

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);

    memset(dp, -1, sizeof dp);

    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:9:10: 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]
    9 |     if(i == a.size()) {
      |        ~~^~~~~~~~~~~

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