Submission #1118719

#	Time	Username	Problem	Language	Result	Execution time	Memory
1118719		Zero_OP	Rainforest Jumps (APIO21_jumps)	C++14	0 / 100	4 ms	608 KiB

jumps

#include <bits/stdc++.h>

using namespace std;

const int MAX = 2e3 + 5;
const int inf = 1e9;

int N, H[MAX];
vector<int> adj[MAX];

struct RMQ{
    vector<vector<int>> rmq;

    RMQ(vector<int>& a) : rmq(rmq){
        int N = (int)a.size();
        int L = 32 - __builtin_clz(N);
        rmq.emplace_back(a);
        for(int i = 1; i < L; ++i){
            rmq.emplace_back(N - (1 << i) + 1);
            for(int j = 0; j + (1 << i) <= N; ++j){
                rmq[i][j] = min(rmq[i - 1][j], rmq[i - 1][j + (1 << (i - 1))]);
            }
        }
    }

    int query(int l, int r){
        int k = 31 - __builtin_clz(r - l + 1);
        return min(rmq[k][l], rmq[k][r - (1 << k) + 1]);
    }
};

vector<RMQ> rmqs;

vector<int> bfs(int s){
    vector<int> d(N, inf);
    queue<int> q;
    q.push(s);
    d[s] = 0;

    while(!q.empty()){
        int u = q.front(); q.pop();
        for(auto v : adj[u]){
            if(d[v] == inf){
                d[v] = d[u] + 1;
                q.push(v);
            }
        }
    }

    return d;
}

void init(int _N, vector<int> _H){
    N = _N;
    for(int i = 0; i < N; ++i) H[i] = _H[i];

    stack<int> st;
    for(int i = 0; i < N; ++i){
        while(!st.empty() && H[st.top()] <= H[i]) st.pop();
        if(!st.empty()) adj[i].emplace_back(st.top());
        st.push(i);
    }

    stack<int>().swap(st);
    for(int i = N - 1; i >= 0; --i){
        while(!st.empty() && H[st.top()] <= H[i]) st.pop();
        if(!st.empty()) adj[i].emplace_back(st.top());
        st.push(i);
    }

    for(int i = 0; i < N; ++i){
        vector<int> d = bfs(i);
        rmqs.emplace_back(RMQ(d));
    }
}

int minimum_jumps(int a, int b, int c, int d){
    int cur = inf;
    for(int i = a; i <= b; ++i){
        cur = min(cur, rmqs[i].query(c, d));
    }
    return (cur == inf ? -1 : cur);
}

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

    freopen("task.inp", "r", stdin);
    freopen("task.out", "w", stdout);

    int N, Q;
    cin >> N >> Q;

    vector<int> H(N);
    for(int i = 0; i < N; ++i){
        cin >> H[i];
    }

    init(N, H);

    for(int i = 0; i < Q; ++i){
        int a, b, c, d;
        cin >> a >> b >> c >> d;
        cout << minimum_jumps(a, b, c, d) << '\n';
    }

    return 0;
}
#endif //LOCAL

Compilation message (stderr)

jumps.cpp: In constructor 'RMQ::RMQ(std::vector<int>&)':
jumps.cpp:14:5: warning: 'RMQ::rmq' is initialized with itself [-Winit-self]
   14 |     RMQ(vector<int>& a) : rmq(rmq){
      |     ^~~

• Subtask 1: 0 / 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