Submission #425270

#	Time	Username	Problem	Language	Result	Execution time	Memory
425270		zoooma13	The short shank; Redemption (BOI21_prison)	C++14	55 / 100	2060 ms	15168 KiB

prison

#include <bits/stdc++.h>
using namespace std;

#define MAX_N 2000006
#define INF 0x3f3f3f3f

int sz;
pair <int ,int> tree[2*MAX_N];
void bld(vector<int>&arr){
    for(int i=sz; i<(sz<<1); i++)
        tree[i] = make_pair(arr[i-sz] ,i-sz);
    for(int i=sz-1; i; i--)
        tree[i] = min(tree[i<<1] ,tree[i<<1|1]);
}
void upd(int p ,int v){
    for(tree[p+=sz].first = v; p>1; p>>=1)
        tree[p>>1] = min(tree[p] ,tree[p^1]);
}
pair <int ,int> qry(int l ,int r){
    auto res = make_pair(INF ,INF);
    for(l+=sz ,r+=sz+1; l<r; l>>=1 ,r>>=1){
        if(l&1) res = min(res ,tree[l++]);
        if(r&1) res = min(res ,tree[--r]);
    }
    return res;
}

int main()
{
    int n ,k ,T;
    scanf("%d%d%d",&n,&k,&T) ,sz = n+1;
    vector <int> a(n);
    for(int&i : a)
        scanf("%d",&i);

    int ans = 0;
    vector <int> lf(n ,INF);
    vector <pair<int ,int>> q;
    for(int i = 0; i < n; i++){
        while(!q.empty() && q.back().first > T-i)
            q.pop_back();
        if(a[i] <= T){
            while(!q.empty() && q.back().first >= a[i]-i)
                q.pop_back();
            q.push_back({a[i]-i ,i});
        }else if(!q.empty())
            lf[i-1] = q.back().second;
        else
            ans++;
    }

    bld(lf);
    for(int s = 0; s < k; s++){
        vector <int> cum(n+1);
        for(int i = 0; i < n; i++) if(lf[i] != INF)
            cum[lf[i]]++ ,cum[i+1]--;
        for(int i = 0; i <= n; i++)
            cum[i] += (i? cum[i-1] : 0);

        int rem = max_element(cum.begin() ,cum.end()) - cum.begin();

        while(true){
            auto q = qry(rem ,n-1);
            if(q.first > rem)
                break;
            lf[q.second] = INF;
            upd(q.second ,INF);
            ans++;
        }
    }
    printf("%d\n",n-ans);
}

Compilation message (stderr)

prison.cpp: In function 'int main()':
prison.cpp:31:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   31 |     scanf("%d%d%d",&n,&k,&T) ,sz = n+1;
      |     ~~~~~^~~~~~~~~~~~~~~~~~~
prison.cpp:34:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   34 |         scanf("%d",&i);
      |         ~~~~~^~~~~~~~~

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