Submission #1113134

#TimeUsernameProblemLanguageResultExecution timeMemory
1113134ortsacRainforest Jumps (APIO21_jumps)C++17
33 / 100
4062 ms14136 KiB
#include "jumps.h"
#include <bits/stdc++.h>

using namespace std;

#define int long long
#define pii pair<long long, long long>
#define fr first
#define se second

int n;
vector<int> h;
vector<int> l;
vector<int> r;

void init(int32_t N, vector<int32_t> H) {
  n = N;
  h.resize(n);
  for (int i = 0; i < n; i++) h[i] = H[i];
  l.resize(n);
  r.resize(n);
  stack<pii> s;
  for (int i = 0; i < n; i++) {
    while (!s.empty() && (s.top().fr < h[i])) s.pop();
    if (s.empty()) l[i] = -1;
    else l[i] = s.top().se;
    s.push({h[i], i});
  }
  while (!s.empty()) s.pop();
  for (int i = n - 1; i >= 0; i--) {
    while (!s.empty() && (s.top().fr < h[i])) s.pop();
    if (s.empty()) r[i] = -1;
    else r[i] = s.top().se;
    s.push({h[i], i});
  }
  //for (int i = 0; i < n; i++) cout << l[i] << " " << r[i] << "\n";
  return;
}

const int INF = 0x3f3f3f3f;

vector<int> dist;

int calc(int node) {
  //cout << "node " << node << "\n";
  if (node == -1) return INF;
  if (dist[node] != INF) return dist[node];
  dist[node] = min(calc(l[node]) + 1, calc(r[node]) + 1);
  return dist[node];
}

int32_t minimum_jumps(int32_t A, int32_t B, int32_t C, int32_t D) {
  dist.clear();
  dist.resize(n, INF);
  for (int i = C; i <= D; i++) dist[i] = 0;
  int ans = INF;
  for (int i = A; i <= B; i++) {
    ans = min(ans, calc(i));
  }
  if (ans == INF) return -1;
  else return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...