oj.uz

Submission #491086

# Submission time Handle Problem Language Result Execution time Memory
491086 2021-11-30T13:25:47 Z Newtech66 K blocks (IZhO14_blocks) C++17
0 / 100
 1 ms 1228 KB 
#include<bits/stdc++.h>
using namespace std;
using lol=long long int;
#define endl "\n"
const lol mod1=1e9+7,mod2=998244353;
const lol inf=1e18+8;
const double eps=1e-12;
const int N=2*1e5+5;
#include <ext/pb_ds/assoc_container.hpp> // Common file
#include <ext/pb_ds/tree_policy.hpp> // Including tree_order_statistics_node_update
using namespace __gnu_pbds;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> ordered_set;
//mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int lg2[N];

int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
lg2[1]=0;
for(int i=2;i<N;i++)    lg2[i]=lg2[i/2]+1;
int _=1;
//cin>>_;
while(_--)
{
    int n,k;
    cin>>n>>k;
    vector<vector<lol> > dp(n,vector<lol>(k+1,inf));
    vector<lol> v(n);
    for(int i=0;i<n;i++)
    {
        cin>>v[i];
    }
    vector<int> psg(n,-1);
    stack<lol> st;
    for(int i=0;i<n;i++)
    {
        while(!st.empty() && st.top()<=v[i])    st.pop();
        if(!st.empty())  psg[i]=st.top();
        st.push(i);
    }
    lol mx=0;
    for(int i=0;i<n;i++)
    {
        mx=max(mx,v[i]);
        dp[i][1]=mx;
    }
    for(int l=2;l<=k;l++)
    {
        lol spt[n][25];
        for(int i=0;i<n;i++)
        {
            spt[i][0]=dp[i][l-1];
        }
        for(int j=1;j<25;j++)
        {
            for(int i=0;i+(1<<j)<=n;i++)
            {
                spt[i][j]=min(spt[i][j-1],spt[i+(1<<(j-1))][j-1]);
            }
        }
        auto getmin=[&](int L,int R)
        {
            if(L>R) return inf;
            int j=lg2[R-L+1];
            return min(spt[L][j],spt[R-(1<<j)+1][j]);
        };
        for(int i=0;i<n;i++)
        {
            int p=psg[i];
            if(p==-1)
            {
                dp[i][l]=v[i]+getmin(0,i-1);
            }else
            {
                lol mn=v[i]+getmin(p,i-1);
                mn=min(mn,dp[p][l]);
                dp[i][l]=min(mn,inf);
            }
        }
    }
    /*for(int i=1;i<=k;i++)
    {
        for(int j=0;j<n;j++)
        {
            cout<<dp[j][i]<<" ";
        }
        cout<<endl;
    }*/
    cout<<dp[n-1][k];
}
return 0;
}

Subtask #1
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 972 KB Output is correct
2 Correct 1 ms 972 KB Output is correct
3 Correct 1 ms 1100 KB Output is correct
4 Correct 1 ms 972 KB Output is correct
5 Correct 1 ms 972 KB Output is correct
6 Correct 1 ms 1000 KB Output is correct
7 Correct 1 ms 1100 KB Output is correct
8 Correct 1 ms 972 KB Output is correct
9 Correct 1 ms 1228 KB Output is correct
10 Correct 1 ms 972 KB Output is correct
11 Correct 1 ms 972 KB Output is correct
12 Correct 1 ms 1100 KB Output is correct
13 Incorrect 1 ms 1100 KB Output isn't correct
14 Halted 0 ms 0 KB -

Subtask #2
0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 972 KB Output is correct
2 Correct 1 ms 972 KB Output is correct
3 Correct 1 ms 972 KB Output is correct
4 Correct 1 ms 972 KB Output is correct
5 Correct 1 ms 972 KB Output is correct
6 Correct 1 ms 1100 KB Output is correct
7 Correct 1 ms 972 KB Output is correct
8 Correct 1 ms 972 KB Output is correct
9 Correct 1 ms 972 KB Output is correct
10 Correct 1 ms 1100 KB Output is correct
11 Correct 1 ms 1100 KB Output is correct
12 Incorrect 1 ms 972 KB Output isn't correct
13 Halted 0 ms 0 KB -

Subtask #3
0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 972 KB Output is correct
2 Correct 1 ms 972 KB Output is correct
3 Correct 1 ms 1100 KB Output is correct
4 Correct 1 ms 972 KB Output is correct
5 Correct 1 ms 972 KB Output is correct
6 Correct 1 ms 1000 KB Output is correct
7 Correct 1 ms 1100 KB Output is correct
8 Correct 1 ms 972 KB Output is correct
9 Correct 1 ms 1228 KB Output is correct
10 Correct 1 ms 972 KB Output is correct
11 Correct 1 ms 972 KB Output is correct
12 Correct 1 ms 1100 KB Output is correct
13 Incorrect 1 ms 1100 KB Output isn't correct
14 Halted 0 ms 0 KB -

Subtask #4
0 / 47.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 972 KB Output is correct
2 Correct 1 ms 972 KB Output is correct
3 Correct 1 ms 1100 KB Output is correct
4 Correct 1 ms 972 KB Output is correct
5 Correct 1 ms 972 KB Output is correct
6 Correct 1 ms 1000 KB Output is correct
7 Correct 1 ms 1100 KB Output is correct
8 Correct 1 ms 972 KB Output is correct
9 Correct 1 ms 1228 KB Output is correct
10 Correct 1 ms 972 KB Output is correct
11 Correct 1 ms 972 KB Output is correct
12 Correct 1 ms 1100 KB Output is correct
13 Incorrect 1 ms 1100 KB Output isn't correct
14 Halted 0 ms 0 KB -