Submission #879867

#	Time	Username	Problem	Language	Result	Execution time	Memory
879867		12345678	K blocks (IZhO14_blocks)	C++17	53 / 100	1038 ms	41484 KiB

blocks

#include <bits/stdc++.h>

using namespace std;
#pragma GCC optimize(" unroll-loops")
const int nx=1e5+5, kx=105;
int n, k, v[nx], dp[kx][nx], x, mx;

struct segtree
{
    int d[4*nx], lz[4*nx];
    void pushlz(int l, int r, int i)
    {
        d[i]+=lz[i];
        if (l==r) return void(lz[i]=0);
        d[2*i]+=lz[i];
        d[2*i+1]+=lz[i];
        lz[i]=0;
    }
    void build(int l, int r, int i, int ly)
    {
        lz[i]=0;
        if (l==r) return void(d[i]=l<=1?1e9:dp[ly][l-1]);
        int md=(l+r)/2;
        build(l, md, 2*i, ly);
        build(md+1, r ,2*i+1, ly);
        d[i]=min(d[2*i], d[2*i+1]);
    }
    void update(int l, int r, int i, int ql, int qr, int vl)
    {
        pushlz(l, r, i);
        if (qr<l||r<ql) return;
        if (ql<=l&&r<=qr) return d[i]+=vl, pushlz(l, r, i), void();
        int md=(l+r)/2;
        update(l, md, 2*i, ql, qr, vl);
        update(md+1, r, 2*i+1, ql, qr, vl);
        d[i]=min(d[2*i], d[2*i+1]);
    }
    int query(int l, int r, int i, int ql, int qr)
    {
        pushlz(l, r, i);
        if (r<ql||qr<l) return INT_MAX;
        if (ql<=l&&r<=qr) return d[i];
        int md=(l+r)/2;
        return min(query(l, md, 2*i, ql, qr), query(md+1, r, 2*i+1, ql, qr));
    }
} s;

int main()
{
    cin.tie(NULL)->sync_with_stdio(false);
    cin>>n>>k;
    v[0]=1e9;
    for (int i=1; i<=n; i++) cin>>v[i], mx=max(mx, v[i]), dp[1][i]=mx;
    for (int i=2; i<=k; i++)
    {
        stack<int> st;
        st.push(0);
        s.build(1, n, 1, i-1);
        for (int j=1; j<=n; j++)
        {
            while (st.size()>=2&&v[st.top()]<=v[j]) x=st.top(), st.pop(), s.update(1, n, 1, st.top()+1, x, -v[x]);
            s.update(1, n, 1, st.top()+1, j, v[j]);
            dp[i][j]=s.query(1, n, 1, 1, j);
            st.push(j);
        }
    }
    cout<<dp[k][n];
}

/*
5 2
1 2 3 4 5

5 1
1 2 3 4 5

7 3
10 3 5 7 2 3 1
*/

Compilation message (stderr)

blocks.cpp:4:37: warning: bad option '-f unroll-loops' to pragma 'optimize' [-Wpragmas]
    4 | #pragma GCC optimize(" unroll-loops")
      |                                     ^
blocks.cpp:11:36: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   11 |     void pushlz(int l, int r, int i)
      |                                    ^
blocks.cpp:19:43: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   19 |     void build(int l, int r, int i, int ly)
      |                                           ^
blocks.cpp:28:60: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   28 |     void update(int l, int r, int i, int ql, int qr, int vl)
      |                                                            ^
blocks.cpp:38:50: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   38 |     int query(int l, int r, int i, int ql, int qr)
      |                                                  ^
blocks.cpp:48:10: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   48 | int main()
      |          ^

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