Submission #903590

#	Time	Username	Problem	Language	Result	Execution time	Memory
903590		efedmrlr	Rainforest Jumps (APIO21_jumps)	C++17	23 / 100	865 ms	49092 KiB

jumps

#include "jumps.h"

// #pragma GCC optimize("O3,Ofast,unroll-loops")
// #pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/stdc++.h>

using namespace std;


#define lli long long int
#define MP make_pair
#define pb push_back
#define REP(i,n) for(int i = 0; (i) < (n); (i)++)
#define all(x) x.begin(), x.end()


void fastio() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
}


const double EPS = 0.00001;
const int INF = 1e9+500;
const int ALPH = 26;
const int LGN = 25;
const int MAXN = 2e5+5;
constexpr int MOD = 1e9+7;
int n,m,q;
vector<int> h(MAXN), revh(MAXN);
vector<int> low[LGN], high[LGN];
vector<array<int,2> > adj(MAXN, array<int,2>({0,0}));

void init(int N, std::vector<int> H) {
    h[0] = 0;
    n = N;
    for(int i = 1 ;i<=n; i++) {
        h[i] = H[i - 1];
    }
    for(int i = 1; i<=n; i++) {
        revh[h[i]] = i;
    }
    set<int> st;
    vector<array<int,2> > srt(n);
    for(int i = 1; i<=n; i++) {
        srt[i - 1] = {h[i], i};
        st.insert(i);
    }    
    sort(srt.begin(), srt.end());
    for(auto c : srt) {
        st.erase(c[1]);
        auto itr = st.lower_bound(c[1]);
        if(st.empty()) break;
        if(itr != st.end()) {
            adj[c[1]][1] = *itr;
        }
        if(itr != st.begin()) {
            itr--;
            adj[c[1]][0] = *itr;
        }
        if(h[adj[c[1]][0]] > h[adj[c[1]][1]]) {
            swap(adj[c[1]][0], adj[c[1]][1]);
        }
    }
    REP(i, LGN) {
        low[i].assign(n + 1, 0);
        high[i].assign(n + 1, 0);
    }
    for(int i = 1; i<=n; i++) {
        low[0][i] = adj[i][0];
        high[0][i] = adj[i][1];
    }
    for(int k = 1; k<LGN; k++) {
        for(int i = 1; i <= n; i++) {
            low[k][i] = low[k - 1][low[k - 1][i]];
            high[k][i] = high[k - 1][high[k - 1][i]];
        }
    }

}

int solve(int A, int C) {
    if(h[A] > h[C]) return -1;
    int ans = 0;
    for(int i = LGN - 1; i>=0; i--) {
        if(h[high[i][A]] && h[high[i][A]] <= h[C]) {
            A = high[i][A];
            ans += 1<<i;
        }
    }
    for(int i = LGN - 1; i>=0; i--) {
        if(h[low[i][A]] && h[low[i][A]] <= h[C]) {
            A = low[i][A];
            ans += 1<<i;
        }
    }
    // cout<<"ans:"<<ans<<" A:"<<A<<" C:"<<C<<"\n";
    if(A == C) return ans;
    return -1;
    
}

int minimum_jumps(int A, int B, int C, int D) {
    A++; B++; C++; D++;
    return solve(A, C);
}

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