Submission #769572

#	Time	Username	Problem	Language	Result	Execution time	Memory
769572		dxz05	Rainforest Jumps (APIO21_jumps)	C++17	4 / 100	1002 ms	73920 KiB

jumps

#pragma GCC optimize("Ofast,O3,unroll-loops")
#pragma GCC target("avx,avx2")

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

using namespace std;

const int MaxN = 200000;
const int Log = 18;

int N;
int low[MaxN][Log], high[MaxN][Log];
int lf[MaxN][Log], rg[MaxN][Log];
int rmq[MaxN][Log];

vector<int> H;

void init(int _N, vector<int> _H) {
    N = _N, H = _H;

    stack<int> st;
    for (int i = 0; i < N; i++){
        while (!st.empty() && H[st.top()] < H[i]) st.pop();
        lf[i][0] = (st.empty() ? -1 : st.top());
        st.push(i);
    }

    while (!st.empty()) st.pop();

    for (int i = N - 1; i >= 0; i--){
        while (!st.empty() && H[st.top()] < H[i]) st.pop();
        rg[i][0] = (st.empty() ? -1 : st.top());
        st.push(i);
    }

    memset(low, -1, sizeof(low));
    memset(high, -1, sizeof(high));

    for (int i = 0; i < N; i++){
        int l = lf[i][0], r = rg[i][0];
        if (l == -1 && r == -1) continue;

        if (l != -1 && r != -1){
            low[i][0] = (H[l] < H[r] ? l : r);
            high[i][0] = (H[l] > H[r] ? l : r);
        } else {
            low[i][0] = high[i][0] = (l != -1 ? l : r);
        }

    }

    for (int j = 1; j < Log; j++){
        for (int i = 0; i < N; i++){
            if (low[i][j - 1] != -1){
                low[i][j] = low[low[i][j - 1]][j - 1];
            }

            if (high[i][j - 1] != -1){
                high[i][j] = high[high[i][j - 1]][j - 1];
            }

            if (lf[i][j - 1] != -1){
                lf[i][j] = lf[lf[i][j - 1]][j - 1];
            } else {
                lf[i][j] = -1;
            }

            if (rg[i][j - 1] != -1){
                rg[i][j] = rg[rg[i][j - 1]][j - 1];
            } else {
                rg[i][j] = -1;
            }
        }
    }

    for (int i = 0; i < N; i++) rmq[i][0] = H[i];
    for (int j = 1; j < Log; j++){
        for (int i = 0; i + (1 << j) - 1 < N; i++){
            rmq[i][j] = max(rmq[i][j - 1], rmq[i + (1 << (j - 1))][j - 1]);
        }
    }

}

int range_max(int l, int r){
    int z = 31 - __builtin_clz(r  - l + 1);
    return max(rmq[l][z], rmq[r - (1 << z) + 1][z]);
}

int minimum_jumps(int A, int B, int C, int D) {
    int dest_max = range_max(C, D);

    if (H[B] > dest_max) return -1;
    if (C == B - 1) return 1;

    int ans = 0;
    int v = B;

    int middle = range_max(B + 1, C - 1);
    if (middle > dest_max) return -1;

    for (int i = Log - 1; i >= 0; i--){
        int u = lf[v][i];
        if (u != -1 && u >= A && H[u] <= dest_max){
            v = u;
        }
    }

    for (int i = Log - 1; i >= 0; i--){
        int u = high[v][i];
        if (u != -1 && H[u] <= middle){
            v = u;
            ans += 1 << i;
        }
    }

    for (int i = Log - 1; i >= 0; i--){
        int u = rg[v][i];
        if (u != -1 && u < C){
            ans += 1 << i;
            v = u;
        }
    }

    if (v < C){
        v = rg[v][0];
        ans++;
    }

    return (C <= v && v <= D ? ans : -1);
}

• Subtask 1: 4 / 4
• Subtask 2: 0 / 8
• Subtask 3: 0 / 13
• Subtask 4: 0 / 12
• Subtask 5: 0 / 23
• Subtask 6: 0 / 21
• Subtask 7: 0 / 19