Submission #647677

#	Time	Username	Problem	Language	Result	Execution time	Memory
647677		sugartheanh	K개의 묶음 (IZhO14_blocks)	C++14	0 / 100	74 ms	164812 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] = 1e18;
    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])
            {
                if(i <= j)
                {
                    dp[i][j] = min(dp[i][j], f[i-1][st.top()] + a[j]);
                    tmp = min(tmp, f[i][st.top()]);
                }
                st.pop();
            }
            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];
            }
            f[i][st.top()] = min(f[i][st.top()], tmp);
            st.push(j);
//            cout<<i<<' '<<j<<' '<<dp[i][j]<<'\n';
        }
    }
    cout<<dp[k][n];
}

Compilation message (stderr)

blocks.cpp: In function 'int main()':
blocks.cpp:27:12: warning: overflow in conversion from 'double' to 'int' changes value from '1.0e+18' to '2147483647' [-Woverflow]
   27 |     a[0] = 1e18;
      |            ^~~~

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