Submission #1025957

#	Time	Username	Problem	Language	Result	Execution time	Memory
1025957		ttamx	The short shank; Redemption (BOI21_prison)	C++17	100 / 100	228 ms	49512 KiB

prison

#include<bits/stdc++.h>

using namespace std;

const int INF=INT_MAX/2;

const int N=2e6+5;

int n,d,t,ans;
int a[N],pre[N],par[N],dep[N],deep[N],cnt[N];
stack<int> s,st;

int main(){
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n >> d >> t;
    for(int i=1;i<=n;i++)cin >> a[i];
    a[0]=-INF;
    s.emplace(0);
    for(int i=1;i<=n;i++){
        int x=min(a[i],t)-i;
        while(a[s.top()]-s.top()>x)s.pop();
        if(a[i]>t)pre[i]=s.top();
        else ans++;
        s.emplace(i);
        if(pre[i]){
            ans++;
            while(!st.empty()&&pre[st.top()]>=pre[i]){
                par[st.top()]=i;
                st.pop();
            }
            st.emplace(i);
        }
    }
    for(int i=1;i<=n;i++)if(pre[i]){
        cnt[dep[deep[i]]]--;
        dep[i]=dep[deep[i]]+1;
        cnt[dep[i]]++;
        if(dep[i]>dep[deep[par[i]]])deep[par[i]]=i;
    }
    for(int i=n;i>=1;i--)while(cnt[i]--&&d-->0)ans-=i;
    cout << ans;
}

• Subtask 1: 15 / 15
• Subtask 2: 10 / 10
• Subtask 3: 20 / 20
• Subtask 4: 10 / 10
• Subtask 5: 25 / 25
• Subtask 6: 20 / 20