Submission #805472

#	Time	Username	Problem	Language	Result	Execution time	Memory
805472		QwertyPi	Dancing Elephants (IOI11_elephants)	C++14	26 / 100	12 ms	1156 KiB

elephants

#include "elephants.h"
#include <bits/stdc++.h>

const int MAXN = 1.5e5 + 11;
const int SQ = 411;
int n, l;
int x[MAXN];

using namespace std;

struct naive{
    int n, y[SQ * 2], dp[SQ * 2], fin[SQ * 2];
    void build(int n, int x[]){
        naive::n = n;
        for(int i = 0; i < n; i++){
            y[i] = x[i];
        }
    }
    void rem(int x){
        int i; for(i = 0; i < n; i++) if(y[i] == x) break;
        for(int j = i; j < n - 1; j++) y[j] = y[j + 1];
        n--; precalc();
    }
    void add(int x){
        int i; for(i = 0; i < n; i++) if(y[i] > x) break;
        for(int j = n - 1; j >= i; j--) y[j + 1] = y[j];
        y[i] = x;
        n++; precalc();
    }
    void precalc(){
        dp[n] = 0; fin[n] = n; int r = n - 1;
        for(int j = n - 1; j >= 0; j--){
            while(y[r] - y[j] > l) r--;
            dp[j] = dp[r + 1] + 1;
            fin[j] = r + 1 != n ? fin[r + 1] : j;
        }
    }
    pair<int, int> qry(int x){
        int id = upper_bound(y, y + n, x) - y;
        return {dp[id], y[fin[id]]};
    }
} cal[MAXN / SQ + 10];

int B[MAXN / SQ + 10], size_B;
void rebuild(){
    std::sort(x, x + n);
    B[0] = 0;
    for(int i = 1; i < (n + SQ - 1) / SQ; i++){
        B[i] = x[SQ * i];
    }
    B[(n + SQ - 1) / SQ] = 1 << 30;
    size_B = (n + SQ - 1) / SQ;
    for(int i = 0; i < (n + SQ - 1) / SQ; i++){
        cal[i].build(min(SQ, n - i * SQ), x + i * SQ);
    }
}

void rem(int x){
    int id = upper_bound(B, B + size_B, x) - B - 1;
    cal[id].rem(x);
}

void add(int x){
    int id = upper_bound(B, B + size_B, x) - B - 1;
    cal[id].add(x);
}

int qry(){
    int ans = 0, x = -1, b = 0;
    for(int b = 0; b < size_B; b++){
        if(x >= B[b + 1]) continue;
        pair<int, int> p = cal[b].qry(x);
        ans += p.first; if(p.first != 0) x = p.second + l;
    }
    return ans;
}

void init(int N, int L, int X[]){
    n = N; l = L; for(int i = 0; i < n; i++) x[i] = X[i]; rebuild();
}


int update(int i, int y){
    rem(x[i]);
    x[i] = y;
    add(x[i]);
    return qry();
}

Compilation message (stderr)

elephants.cpp: In function 'int qry()':
elephants.cpp:69:26: warning: unused variable 'b' [-Wunused-variable]
   69 |     int ans = 0, x = -1, b = 0;
      |                          ^

• Subtask 1: 10 / 10
• Subtask 2: 16 / 16
• Subtask 3: 0 / 24
• Subtask 4: 0 / 47
• Subtask 5: 0 / 3