제출 #1344205

#제출 시각아이디문제언어결과실행 시간메모리
1344205Jakub_WozniakThe short shank; Redemption (BOI21_prison)C++20
55 / 100
1571 ms589824 KiB
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define st first
#define nd second
const ll maxn = 500000+912;
vector<vector<int>> dp;
int t[maxn];
int n , D , T;

const int base = (1<<19);
vector <pii> drz[base*2]; // {czas , wartosc w danym czasie}
int aktTIME = 1;

void nowy(int pos , int T , int w)
{
    if(drz[pos].back().nd >= w)return ;
    drz[pos].push_back({T,w});
}

void U(int l , int r, int a ,int b , int pos)
{
    if((drz[pos].back()).nd == r-l+1 && pos < base) // wszystkie zapalone
    {
        nowy(pos*2,drz[pos].back().st , drz[pos].back().nd/2);
        nowy(pos*2+1,drz[pos].back().st , drz[pos].back().nd/2);
    }

    if(a <= l && r <= b)
    {
        nowy(pos,aktTIME , r-l+1);
    }
    else if(r < a || b < l)return ;
    else
    {
        int s = (l+r)/2;
        U(l,s,a,b,pos*2);
        U(s+1,r,a,b,pos*2+1);
        int w = drz[pos*2].back().nd + drz[pos*2+1].back().nd;
        nowy(pos,aktTIME,w);
    }
}

int ansi = 0;
void Q(int l , int r, int a ,int b , int pos , int TIM)
{
    if((drz[pos].back()).nd == r-l+1 && pos < base) // wszystkie zapalone
    {
        nowy(pos*2,drz[pos].back().st , drz[pos].back().nd/2);
        nowy(pos*2+1,drz[pos].back().st , drz[pos].back().nd/2);
    }

    if(a <= l && r <= b)
    {
        auto it = lower_bound(drz[pos].begin() , drz[pos].end() , (pii){TIM+1,-1});
        it--;   
        ansi += (*it).nd;
    }
    else if(r < a || b < l)return ;
    else
    {
        int s = (l+r)/2;
        Q(l,s,a,b,pos*2 , TIM);
        Q(s+1,r,a,b,pos*2+1, TIM);
    }
}


int maxidrz[base*2];
void Up(int X , int val)
{
    X += base;
    maxidrz[X] = val;
    X /= 2;
    while(X)
    {
        maxidrz[X] = max(maxidrz[X*2] , maxidrz[2*X+1]);
        X /= 2;
    }
}

int Qp(int l , int r)
{
    if(r < l)return 0;
    int res =0;
    l += base;
    r += base;
    l--;
    r++;
    while(l/2 != r/2)
    {
        if((l&1) == 0)res = max(res,maxidrz[l+1]);
        if((r&1) == 1)res = max(res,maxidrz[r-1]);
        l /= 2;
        r /= 2;
    }
    return res;
}

void rek(int i1 , int i2 , int l , int r , int d)
{
    if(i2 < i1)return ;
    int i = (i1+i2)/2;

    dp[i][d] = maxn*10+124;
    int COST; // trzeba zupdatowac bo pomijam [i+1,l-1]
    ansi = 0;
    if(i <= l-2)Q(0,base-1,i,l-2,1,(n-i+1));
    COST = ansi;

    int Pop = 0;
    if(i <= l-2) Pop = Qp(i,l-2) - i - (l-2-i+1) + 1; // trzeba zupdatowac bo pomijam [i+1,l-1]
    
    int opt = -1;
    for(int j = max(l,i+1) ; j <= r ; j++)
    {
        Pop--;
        Pop = max(Pop,t[j-1]);
        if(Pop >= 0)COST++;

        if(dp[i][d] > dp[j][d-1] + COST)opt = j;
        dp[i][d] = min(dp[i][d] , dp[j][d-1] + COST);
    }
    rek(i1,i-1,l,opt,d);
    rek(i+1,i2,opt,r,d);
}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    cin >> n >> D >> T;
    for(int i = 1; i<= n ;i++)
    {
        cin >> t[i];
        t[i] = T-t[i];
    }
    cerr << '\n';

    for(int i = 0 ; i < base*2 ; i++)drz[i].push_back({0,0});
    for(int i = n ; i >= 1 ;i--)
    {
        //n <-> 1
        //n-1 <-> 2
        //...
        // n-i <-> i+1  ----> i  <-> n-i+1 
        if(t[i] >= 0)U(0,base-1,i,min(i+t[i],base-1),1);
        aktTIME++;
        Up(i , t[i]+i);
    }

    dp.resize(n+5);
    for(int i = 0 ; i <= n+3 ; i++)dp[i].resize(D+3);

    for(int i = 1; i <= n ; i++)dp[i][0] = maxn*123+15;
    for(int d = 1 ; d <= D+1 ; d++)
    {
        rek(1,n,1,n+1,d);
    }

    cout << dp[1][D+1] << '\n';

    return 0;
}
#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...