Submission #922287

#	Time	Username	Problem	Language	Result	Execution time	Memory
922287		Arp	Rainforest Jumps (APIO21_jumps)	C++17	Compilation error	0 ms	0 KiB

jumps

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

const int N = 2e5 + 10;
int ml[N],mr[N];
int best[N];
int arr[N];
const int inf = 1e9;
int MAXN;
bool one = true;
void init(int n,vector<int> a){
  for(int i = 1;i<=n;i++){
    ml[i] = 0;
    mr[i] = n + 1;
    arr[i] = a[i - 1];
    one &= (arr[i] == i);
  }  
  MAXN = n;
  arr[0] = arr[n + 1] = -1;
  stack<int> lg;
  for(int i = 1;i<=n;i++){
    while(lg.size() > 0 && arr[lg.top()] < arr[i]) lg.pop();
    if(lg.size() > 0){
      ml[i] = lg.top();
    }
    lg.push(i);
  }
  stack<int> rg;
  for(int i = n;i >= 1;i--){
    while(rg.size() > 0 && arr[rg.top()] < arr[i]) rg.pop();
    if(rg.size() > 0){
      mr[i] = rg.top();
    }
    rg.push(i);
  }
  for(int i = 1;i<=n;i++){
    best[i] = (arr[ml[i]] > arr[mr[i]] ? ml[i] : mr[i]);
    // cout << best[i] << " ";
  }
  // cout << '\n';
}

int minimum_jumps(int a,int b,int c,int d){
  a ++,b ++;
  c ++,d ++;
  // if(one) return c - b;  
  int m2 = c;
  for(int i = c;i <= d;i ++){
    if(arr[i] > arr[m2]) m2 = i;
  }
  int m1 = MAXN + 1;
  for(int i = a;i <= b;i++){
    if(arr[i] > arr[m1] && arr[i] < arr[m2]) m1 = i;
  }
  if(m1 == MAXN + 1){
    assert(!one);
    return -1;
  }
  int last = m1;
  int ans = 0;
  while(arr[best[last]] < arr[m2]){
    if(last >= c && last <= d) return ans;
    if(mr[last] >= c && mr[last] <= d) return ans + 1;
    last = best[last];
    assert(last < d && last != MAXN);
    ans ++;
  }
  while(!(last >= c && last <= d)){
    if(arr[last] > arr[m2] || last > d) return -1;
    last = mr[last];
    ans ++;
  }
  return ans;
}

int main(){

  ios_base::sync_with_stdio(false);
  cin.tie(NULL);
  int n;
  cin >> n;
  vector<int> a(n);
  for(int i = 0;i<n;i++){
    cin >> a[i];
  }
  init(n,a);
  int q;
  cin >> q;
  while(q --){
    int a,b,c,d;
    cin >> a >> b >> c >> d;
    cout << minimum_jumps(a,b,c,d) << '\n';
  }
  return 0;
}

Compilation message (stderr)

/usr/bin/ld: /tmp/ccbj9mCr.o: in function `main':
stub.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccgkehls.o:jumps.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status