Submission #311857

#	Time	Username	Problem	Language	Result	Execution time	Memory
311857		nikatamliani	K blocks (IZhO14_blocks)	C++14	53 / 100	27 ms	35840 KiB

blocks

#include <bits/stdc++.h>
using namespace std;
const int N = 3005, oo = 1e9;
int n, k, a[N], dp[N][N];
int main() {
    cin >> n >> k;
    for(int i = 1; i <= n; ++i) {
        cin >> a[i];
    }
    for(int i = 0; i < N; ++i) {
        for(int j = 0; j < N; ++j) {
            dp[i][j] = oo;
        }
    }
    stack < int > st;
    dp[0][0] = 0;
    vector < int > v[k + 1];
    v[0].push_back(0);
    // for(int i = 0; i <= k; ++i) v[i].push_back(0);
    for(int i = 1; i <= n; ++i) {
        while(!st.empty() && a[st.top()] <= a[i]) st.pop();
        int x = 0;
        if(!st.empty()) x = st.top();
        for(int j = min(i, k); j >= 1; --j) {
            dp[i][j] = dp[x][j];
            int l = 0, r = (int)v[j - 1].size() - 1, ans = -1;
            while(r >= l) {
                int m = l + r >> 1;
                if(v[j - 1][m] >= x) {
                    ans = v[j - 1][m];
                    r = m - 1;
                } else {
                    l = m + 1;
                }
            }
            if(~ans)dp[i][j] = min(dp[ans][j - 1] + a[i], dp[i][j]);
            while(!v[j].empty() && dp[v[j].back()][j] >= dp[i][j]) v[j].pop_back();
            v[j].push_back(i);
        }
        st.push(i);
    }
    cout << dp[n][k] << '\n';
}

Compilation message (stderr)

blocks.cpp: In function 'int main()':
blocks.cpp:28:27: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   28 |                 int m = l + r >> 1;
      |                         ~~^~~

• Subtask 1: 14 / 14
• Subtask 2: 18 / 18
• Subtask 3: 21 / 21
• Subtask 4: 0 / 47