Submission #647686

#	Time	Username	Problem	Language	Result	Execution time	Memory
647686		sugartheanh	K개의 묶음 (IZhO14_blocks)	C++14	100 / 100	306 ms	165916 KiB

blocks

#include <bits/stdc++.h>
#define ll long long
#define fo(i,a,b) for(int i=a;i<=b;++i)
#define fod(i,a,b) for(int i=a;i>=b;--i)
#define ii pair<ll,ll>
#define iii pair<ll,ii>
#define fi first
#define se second
#define oo 1e18
#define bit(x,i) (((x)>>(i))&1)
using namespace std;
const int MOD = 1e9 + 7;
const int N = 1e5 + 5;
const int K = 1e2 + 5;
int n,k,a[N];
ll dp[K][N],f[K][N];
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    cin>>n>>k;
    fo(i,1,n)
        cin>>a[i];
    memset(dp,0x3f,sizeof(dp));
    memset(f,0x3f,sizeof(f));
    dp[0][0] = 0;
    a[0] = 1e9;
    fo(i,1,k)
    {
        stack<int>st;
        st.push(0);
        int mx = 0;
        fo(j,1,n)
        {
            mx = max(mx, a[j]);
            if(i == 1)
                dp[i][j] = f[i][j] = mx;
            ll tmp = 1e18;
            while(!st.empty() && a[st.top()] <= a[j])
            {
                int last = st.top();
                st.pop();
                f[i-1][st.top()] = min(f[i-1][st.top()], f[i-1][last]);
            }
            if(i <= j)
            {
                dp[i][j] = min({dp[i][j], dp[i][st.top()], f[i-1][st.top()] + a[j]});
                f[i][j] = dp[i][j];
            }
            st.push(j);
        }
    }
    cout<<dp[k][n];
}

Compilation message (stderr)

blocks.cpp: In function 'int main()':
blocks.cpp:38:16: warning: unused variable 'tmp' [-Wunused-variable]
   38 |             ll tmp = 1e18;
      |                ^~~

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