Submission #797762

# Submission time Handle Problem Language Result Execution time Memory
797762 2023-07-29T21:19:52 Z Blagoj Dancing Elephants (IOI11_elephants) C++17
97 / 100
9000 ms 8396 KB
#include <bits/stdc++.h>
#include "elephants.h"

using namespace std;

int n, L;
const int K = 400;

int block[K][2 * K], nxt[K][2 * K], val[K][2 * K], sz[K], b[(int) 2e5+5], a[(int) 2e5+5];
pair<int, int> go[K][2 * K];

void make(int k) {
    int r = 0;
    for (int i = 0; i < sz[k]; i++) {
        while (r < sz[k] && block[k][r] - block[k][i] <= L) r++;
        nxt[k][i] = r;
        val[k][i] = block[k][i] + L;
    }
    for (int i = sz[k] - 1; i >= 0; i--) {
        if (nxt[k][i] == sz[k]) go[k][i] = {1, val[k][i]};
        else go[k][i] = {1 + go[k][nxt[k][i]].first, go[k][nxt[k][i]].second};
    }
}

void build() {
    n = 0;
    for (int i = 0; i < K; i++)
        for (int j = 0; j < sz[i]; j++) b[n++] = block[i][j];
    for (int i = 0; i < n; i += K) {
        int r = min(n, i + K);
        sz[i / K] = r - i;
        for (int j = 0; j < r; j++) block[i / K][j] = b[i + j];
        make(i / K);
    }
}

void init(int N, int _L, int X[]) {
    n = N, L = _L;
    for (int i = 0; i < n; i++) b[i] = a[i] = X[i];
    for (int i = 0; i < n; i += K) {
        int r = min(n, i + K);
        sz[i / K] = r - i;
        for (int j = 0; j < r; j++) block[i / K][j] = b[i + j];
    }
    build();
}

void erase(int i, int x) {
    for (int j = 0; j < sz[i]; j++) {
        if (block[i][j] < x) continue;
        block[i][j] = block[i][j + 1];
    }
    sz[i]--;
    make(i);
}

void del(int x) {
    for (int i = 0; i < K; i++) {
        if (x <= block[i][sz[i] - 1]) {
            erase(i, x);
            break;
        }
    }
}

void add(int i, int x) {
    int dummy = x;
    for (int j = 0; j < sz[i]; j++) {
        if (block[i][j] < x) continue;
        swap(block[i][j], dummy);
    }
    block[i][sz[i]++] = dummy;
    make(i);
}

void Update(int x) {
    for (int i = 0; i < K; i++) {
        if (x <= block[i][sz[i] - 1]) {
            add(i, x);
            break;
        }
        if (i == (n - 1) / K) {
            add(i, x);
            break;
        }
    }
}

int q = 0;

int update(int x, int y) {
    del(a[x]);
    a[x] = y;
    Update(a[x]);
    q++;
    if (q == K) {
        build();
        q = 0;
    }
    int ans = 0, pos = block[0][0], i = 0, j = 0;
    int cnt = 0;
    for (; sz[i];) {
        auto it = go[i][j];
        ans += it.first;
        pos = it.second;
        while (sz[i] && block[i][sz[i] - 1] <= pos) i++;
        if (!sz[i]) break;
        int l = 0, r = sz[i] - 1;
        while (l < r) {
            int m = (l + r) >> 1;
            if (block[i][m] <= pos) l = m + 1;
            else r = m;
        }
        j = r;
    }
    return ans;
}

Compilation message

elephants.cpp: In function 'int update(int, int)':
elephants.cpp:101:9: warning: unused variable 'cnt' [-Wunused-variable]
  101 |     int cnt = 0;
      |         ^~~
# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 1 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 1 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
7 Correct 267 ms 1620 KB Output is correct
8 Correct 248 ms 1876 KB Output is correct
9 Correct 282 ms 3660 KB Output is correct
10 Correct 302 ms 3668 KB Output is correct
11 Correct 327 ms 3636 KB Output is correct
12 Correct 497 ms 3668 KB Output is correct
13 Correct 287 ms 3596 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 1 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
7 Correct 267 ms 1620 KB Output is correct
8 Correct 248 ms 1876 KB Output is correct
9 Correct 282 ms 3660 KB Output is correct
10 Correct 302 ms 3668 KB Output is correct
11 Correct 327 ms 3636 KB Output is correct
12 Correct 497 ms 3668 KB Output is correct
13 Correct 287 ms 3596 KB Output is correct
14 Correct 266 ms 2132 KB Output is correct
15 Correct 379 ms 2388 KB Output is correct
16 Correct 800 ms 3896 KB Output is correct
17 Correct 866 ms 4992 KB Output is correct
18 Correct 906 ms 4988 KB Output is correct
19 Correct 467 ms 4984 KB Output is correct
20 Correct 858 ms 4992 KB Output is correct
21 Correct 810 ms 5036 KB Output is correct
22 Correct 508 ms 4988 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 0 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
4 Correct 1 ms 340 KB Output is correct
5 Correct 1 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
7 Correct 267 ms 1620 KB Output is correct
8 Correct 248 ms 1876 KB Output is correct
9 Correct 282 ms 3660 KB Output is correct
10 Correct 302 ms 3668 KB Output is correct
11 Correct 327 ms 3636 KB Output is correct
12 Correct 497 ms 3668 KB Output is correct
13 Correct 287 ms 3596 KB Output is correct
14 Correct 266 ms 2132 KB Output is correct
15 Correct 379 ms 2388 KB Output is correct
16 Correct 800 ms 3896 KB Output is correct
17 Correct 866 ms 4992 KB Output is correct
18 Correct 906 ms 4988 KB Output is correct
19 Correct 467 ms 4984 KB Output is correct
20 Correct 858 ms 4992 KB Output is correct
21 Correct 810 ms 5036 KB Output is correct
22 Correct 508 ms 4988 KB Output is correct
23 Execution timed out 9019 ms 8396 KB Time limit exceeded
24 Halted 0 ms 0 KB -