Submission #1282271

#	Time	Username	Problem	Language	Result	Execution time	Memory
1282271		quan606303	K blocks (IZhO14_blocks)	C++20	0 / 100	1 ms	576 KiB

blocks

/*
 * @Author: RMQuan 
 * @Date: 2025-10-22 23:48:40 
 * @Last Modified by: RMQuan
 * @Last Modified time: 2025-10-23 01:54:30
 */
/*idea : 



*/
#include <bits/stdc++.h>
bool M1;
#define ll long long
#define INTMAX INT_MAX
#define INTMIN INT_MIN
#define LONGMAX LLONG_MAX
#define LONGMIN LLONG_MIN
#define fi first
#define se second
#define memfull(a,b) memset(a,b,sizeof(a));
#define endl '\n'
#define TASK "TEST"
#define file() if (fopen(TASK".inp","r")){freopen(TASK".inp","r",stdin); freopen(TASK".out","w",stdout);}
using namespace std;
const int MOD=1e9+7;
const int maxn=1e5+7;
const int LOG=16;
const int inf=1e9;
int a[maxn],n,k;
int rmq[maxn][LOG+1];
struct query
{
    int pos,val,type;
};
bool cmp(const query &x,const query &y)
{
    if (x.pos!=y.pos)return x.pos<y.pos;
    return x.type<y.type;
}
void build(const vector<pair<int,pair<int,int>>> &qr)
{
    for (int i=0;i<=n;i++) rmq[i][0]=inf;

    vector<query> vt;
    vector<int> vals;
    for (auto &i: qr){
        vt.push_back({i.se.fi, i.fi, 1});
        vt.push_back({i.se.se+1, i.fi, -1});
        vals.push_back(i.fi);
    }
    for (int i=0;i<=n;i++) vt.push_back({i,0,2});
    sort(vt.begin(), vt.end(), cmp);
    sort(vals.begin(), vals.end());
    vals.erase(unique(vals.begin(), vals.end()), vals.end());
    int m = vals.size();
    auto id = [&](int x){ return (int)(lower_bound(vals.begin(), vals.end(), x) - vals.begin()) + 1; };
    vector<int> bit(m+5, inf), cnt(m+5, 0);
    auto update=[&](int i, int val){
        for (;i<=m;i+=i&-i) bit[i]=min(bit[i],val);
    };

    auto erase_val=[&](int val_idx){ 
        cnt[val_idx]--;
    };

    auto insert_val=[&](int val_idx, int val){
        cnt[val_idx]++;
        update(val_idx,val);
    };

    auto getMin=[&](){
        int res=inf;
        int i=m;
        while(i>0){
            int candidate=bit[i];
            if (candidate!=inf){
                int idx=id(candidate);
                if (cnt[idx]>0) res=min(res,candidate);
            }
            i -= i & -i;
        }
        return res;
    };
    for (auto &q: vt){
        if (q.type==1){
            int idx=id(q.val);
            insert_val(idx,q.val);
        }
        else if (q.type==-1){
            int idx=id(q.val);
            erase_val(idx);
        }
        else{ 
            rmq[q.pos][0]=getMin();
        }
    }
    for (int j=1;j<=LOG;j++)
        for (int i=0;i+(1<<j)-1<=n;i++)
            rmq[i][j]=min(rmq[i][j-1], rmq[i+(1<<(j-1))][j-1]);
}

int get(int l,int r)
{
    int k=__lg(r-l+1);
    return min(rmq[l][k],rmq[r-(1<<k)+1][k]);
}
int L[maxn],R[maxn];
int dp[maxn];
int32_t main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    file();
    cin>>n>>k;
    for (int i=1;i<=n;i++)cin>>a[i];  
    stack<int> st;
    for (int i=1;i<=n;i++)
    {
        while (st.size()&&a[st.top()]<=a[i])st.pop();
        if (st.size())L[i]=st.top()+1;
        else L[i]=1;
        st.push(i);
    }
    while (st.size())st.pop();
    for (int i=n;i>=1;i--)
    {
        while(st.size()&&a[st.top()]<=a[i])st.pop();
        if (st.size())R[i]=st.top()-1;
        else R[i]=n;
        st.push(i);
    }
    vector<pair<int,pair<int,int> > >qr;
    qr.push_back({0,{0,0}});
    build(qr);
    for (int j=1;j<=k;j++)
    {
        qr.clear();
        for (int i=1;i<=n;i++)
        {
            int l=L[i]-1;
            int r=i-1;
            int u=i;
            int v=R[i];
            int val=get(l,r);
            qr.push_back({val+a[i],{u,v}});
            cerr<<val+a[i]<<" "<<u<<" "<<v<<endl;
        }
        build(qr);
    }
    cout<<get(n,n);
    bool M2;
    cerr<<"-------------------------------------------------"<<endl;
    cerr<<"Time : "<<clock()<<" ms"<<endl;
    cerr<<"Memory : "<<abs(&M2-&M1)/1024/1024<<" MB"<<endl;
    cerr<<"-------------------------------------------------"<<endl;
    return 0;
}

Compilation message (stderr)

blocks.cpp: In function 'int32_t main()':
blocks.cpp:24:50: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   24 | #define file() if (fopen(TASK".inp","r")){freopen(TASK".inp","r",stdin); freopen(TASK".out","w",stdout);}
      |                                           ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~
blocks.cpp:115:5: note: in expansion of macro 'file'
  115 |     file();
      |     ^~~~
blocks.cpp:24:81: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   24 | #define file() if (fopen(TASK".inp","r")){freopen(TASK".inp","r",stdin); freopen(TASK".out","w",stdout);}
      |                                                                          ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
blocks.cpp:115:5: note: in expansion of macro 'file'
  115 |     file();
      |     ^~~~

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