Submission #1106278

#TimeUsernameProblemLanguageResultExecution timeMemory
1106278alexddThe short shank; Redemption (BOI21_prison)C++17
80 / 100
2045 ms133028 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int INF = 1e18;
int n,d,lim;
int t[2000005];
bool blocat[2000005];
int tole[2000005];

pair<int,int> aint[4200000];
int lazy[4200000];
void build(int nod, int st, int dr)
{
    if(st==dr)
    {
        aint[nod] = {0,st};
        return;
    }
    int mij=(st+dr)/2;
    build(nod*2,st,mij);
    build(nod*2+1,mij+1,dr);
    aint[nod] = max(aint[nod*2], aint[nod*2+1]);
}
void propagate(int nod)
{
    lazy[nod*2] += lazy[nod];
    lazy[nod*2+1] += lazy[nod];
    aint[nod*2].first += lazy[nod];
    aint[nod*2+1].first += lazy[nod];
    lazy[nod]=0;
}
void upd(int nod, int st, int dr, int le, int ri, int newv)
{
    if(le>ri)
        return;
    if(le==st && dr==ri)
    {
        aint[nod].first += newv;
        lazy[nod] += newv;
        return;
    }
    propagate(nod);
    int mij=(st+dr)/2;
    upd(nod*2,st,mij,le,min(mij,ri),newv);
    upd(nod*2+1,mij+1,dr,max(mij+1,le),ri,newv);
    aint[nod] = max(aint[nod*2], aint[nod*2+1]);
}

signed main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);
    cin>>n>>d>>lim;
    build(1,1,n);
    deque<int> dq;
    int rez=0;
    for(int i=1;i<=n;i++)
    {
        cin>>t[i];
        while(!dq.empty() && t[i] - i <= t[dq.back()] - dq.back())
            dq.pop_back();
        dq.push_back(i);

        while(!dq.empty() && t[dq.back()] - dq.back() > lim - i)
            dq.pop_back();
        if(dq.empty())
            tole[i] = n+1;
        else
            tole[i] = dq.back();
        upd(1,1,n,tole[i],i-1,+1);
        if(tole[i]!=n+1) rez++;
    }
    for(int pas=1;pas<=d;pas++)
    {
        if(aint[1].first==0)
            break;
        rez -= aint[1].first;
        int unde = aint[1].second;
        blocat[unde]=1;
        for(int i=unde+1;i<=n;i++)
        {
            if(tole[i]<=unde)
            {
                upd(1,1,n,tole[i],i-1,-1);
                tole[i] = n+1;
            }
        }
    }
    cout<<rez;
    return 0;
}
/*
pentru fiecare pozitie i ne intereseaza doar cea mai din dreapta pozitie j aflata in stanga ei care respecta
j <= i
t[j] + (i-j) <= lim
t[j] - j <= lim - i


*/
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...