Submission #743493

#	Time	Username	Problem	Language	Result	Execution time	Memory
743493		saayan007	Rainforest Jumps (APIO21_jumps)	C++17	0 / 100	4116 ms	449232 KiB

jumps

#include "bits/stdc++.h"
using namespace std;

#define fr first
#define sc second
#define eb emplace_back
const char nl = '\n';

void __print(int x) {cerr << x;}
void __print(long x) {cerr << x;}
void __print(unsigned x) {cerr << x;}
void __print(unsigned long x) {cerr << x;}
void __print(unsigned long long x) {cerr << x;}
void __print(float x) {cerr << x;}
void __print(double x) {cerr << x;}
void __print(long double x) {cerr << x;}
void __print(char x) {cerr << '\'' << x << '\'';}
void __print(const char *x) {cerr << '\"' << x << '\"';}
void __print(const string &x) {cerr << '\"' << x << '\"';}
void __print(bool x) {cerr << (x ? "true" : "false");} 
 
template<typename T, typename V>
void __print(const pair<T, V> &x) {cerr << '{'; __print(x.first); cerr << ','; __print(x.second); cerr << '}';}
template<typename T>
void __print(const T &x) {int f = 0; cerr << '{'; for (auto &i: x) cerr << (f++ ? "," : ""), __print(i); cerr << "}";}
void _print() {cerr << "]\n";}
template <typename T, typename... V>
void _print(T t, V... v) {__print(t); if (sizeof...(v)) cerr << ", "; _print(v...);}
#ifndef ONLINE_JUDGE
#define dbg(x...) cerr << "LINE(" << __LINE__ << ") -> " <<"[" << #x << "] = ["; _print(x)
#else
#define dbg(x...)
#endif

vector<int> L, R;
map<pair<int, int>, int> dist;
int n;
const int inf = 2e8L + 1;

vector<int> tp;
void init(int N, std::vector<int> H) {
    n = N;

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

    R.resize(n);
    for(int i = n - 1; i >= 0; --i) {
        while(!st.empty() && H[st.top()] < H[i]) st.pop();
        R[i] = (st.empty() ? -1 : st.top());
        st.push(i);
    }
    /* for(int i = 0; i < N; ++i) { */
    /*     cout << i << " : " << L[i] << ' ' << R[i] << nl; */
    /* } */
    tp.resize(n);
    iota(tp.begin(), tp.end(), 0);
    sort(tp.begin(), tp.end(), [&](int left, int right) {
        return H[left] < H[right];
    });
    /* for(int i : tp) cout << i << ' '; cout << nl; */

    for(int i = 0; i < n; ++i) dist[{i, i}] = 0;
    /* for(int x = 0; x < n; ++x) { */
    /*     dist[x][x] = 0; */
    /*     queue<int> q; */
    /*     q.emplace(x); */
    /*     while(!q.empty()) { */
    /*         int y = q.front(); q.pop(); */
    /*         if(L[y] != -1 && dist[x][L[y]] == -1) { */
    /*             dist[x][L[y]] = dist[x][y] + 1; */
    /*             q.emplace(L[y]); */
    /*         } */
    /*         if(R[y] != -1 && dist[x][R[y]] == -1) { */
    /*             dist[x][R[y]] = dist[x][y] + 1; */
    /*             q.emplace(R[y]); */
    /*         } */
    /*     } */
    /*     for(int y = 0; y < n; ++y) { */
    /*         /1* cout << dist[x][y] << ' '; *1/ */
    /*         if(dist[x][y] == -1) dist[x][y] = 10*n; */
    /*     } */
    /*     /1* cout << nl; *1/ */
    /* } */
}

int bfs(int st, int nd) {
    int res = inf;
    queue<int> q;
    q.emplace(st);
    while(!q.empty()) {
        int x = q.front();
        q.pop();
        if(L[x] != -1 && !dist.count({st, L[x]})) {
            dist[{st, L[x]}] = dist[{st, x}] + 1;
            if(dist.count({L[x], nd})) {
                res = min(res, dist[{st, L[x]}] + dist[{L[x], nd}]);
            }
            else q.emplace(L[x]);
            if(L[x] == nd) {
                return dist[{st, nd}];
            }
        }
        if(R[x] != -1 && !dist.count({st, R[x]})) {
            dist[{st, R[x]}] = dist[{st, x}] + 1;
            if(dist.count({R[x], nd})) {
                res = min(res, dist[{st, R[x]}] + dist[{R[x], nd}]);
            }
            else q.emplace(R[x]);
            if(R[x] == nd) {
                return dist[{st, nd}];
            }
        }
    }
    return dist[{st, nd}] = res;
}

int minimum_jumps(int A, int B, int C, int D) {
    int res = bfs(A, C);
    return res > n ? -1 : res;
}

• 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