Submission #918481

#	Time	Username	Problem	Language	Result	Execution time	Memory
918481		n3rm1n	K blocks (IZhO14_blocks)	C++17	0 / 100	1 ms	2396 KiB

blocks

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
const long long MAXN = 1e5 + 10, MAXK = 105;
void speed()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
}
long long n, k, a[MAXN];
void read()
{
    cin >> n >> k;
    for (long long i = 1; i <= n; ++ i)
        cin >> a[i];
}
long long dp[MAXN][MAXK];
void solve()
{
    long long inf = 1e17;
    for (long long j = 1; j <= k; ++ j)
    {
        stack < pair <long long, long long > > t;
        for (long long i = j; i <= n; ++ i)
        {
            dp[i][j] = inf;
            dp[i][j] = dp[i-1][j-1] + a[i];
            long long init = dp[i][j];
            while(!t.empty())
            {
                long long sz = t.top().first;
                long long dpp = t.top().second;

                if(sz < a[i])
                {
                    if(dpp + a[i] - sz < init)dp[i][j] = dpp + a[i] - sz;
                    t.pop();
                }
                else break;
            }
            if(t.size() == 0 || t.top().second > dp[i][j])t.push(make_pair(a[i], dp[i][j]));
            else dp[i][j] = min(dp[i][j], t.top().second);
            //cout << i << " " << j << " --> " << dp[i][j] << endl;
        }
    }
    cout << dp[n][k] << endl;
}
int main()
{
    speed();

    read();
    solve();
    return 0;
}

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