Submission #1119961

#	Time	Username	Problem	Language	Result	Execution time	Memory
1119961		HuyAT	K blocks (IZhO14_blocks)	C++14	100 / 100	148 ms	82736 KiB

blocks

#include<bits/stdc++.h>
#define newl '\n'

const int N = 1e5 + 10;
const int K = 1e2 + 10;
const int V = 1e7 + 10;
const long long INF = 1e18;
const long long M = 1e9 + 7;

long long a[N + 1],n,k;
long long f[K + 1][N + 1];

void readData(){
    std::cin >> n >> k;
    for(int i = 1;i <= n;++i){
        std::cin >> a[i];
    }
}
long long solve(){
    for(int i = 0;i <= k;++i){
        for(int j = 0;j <= n;++j){
            f[i][j] = INF;
        }
    }
    f[0][0] = 0;
    for(int i = 1;i <= k;++i){
        std::stack<std::pair<int,long long>> st;
        for(int j = 1;j <= n;++j){
            long long min_f = f[i - 1][j - 1];
            while(!st.empty() && a[j] >= a[st.top().first]){
                min_f = std::min(min_f,st.top().second);
                st.pop();
            }
            f[i][j] = std::min(f[i][st.empty() ? 0 : st.top().first],min_f + a[j]);
            st.push({j,min_f});
        }
    }
    return f[k][n];
}
int main(){
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);std::cout.tie(nullptr);
    readData();
    std::cout << solve();
    return 0;
}

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