제출 #1355331

#제출 시각아이디문제언어결과실행 시간메모리
1355331JohanSum Zero (RMI20_sumzero)C++20
컴파일 에러
0 ms0 KiB
#include <bits/stdc++.h>
using namespace std;
const int INF = 1e9 + 4;
const int N = 4e5 + 5;
const int LG = 20;
int a[N], c[N], st[N * 4], up[N][LG], dpt[N], mx[N];
vector < int > adj[N];
void build(int v, int l, int r){
  if(l == r){
    st[v] = c[l];
    return;
  }
  int mid = (l + r) >> 1;
  build(v * 2, l, mid);
  build(v * 2 + 1, mid + 1, r);
  st[v] = max(st[v * 2], st[v * 2 + 1]);
}
int ask(int v, int l, int r, int ql, int qr, int x){
  if(ql > r || qr < l || st[v] <= x)
    return -1;
  if(l == r)
    return l;
  int mid = (l + r) >> 1;
  int rs = ask(v * 2, l, mid, ql, qr, x);
  if(rs == -1)
    rs = ask(v * 2 + 1, mid + 1, r, ql, qr, x);
  return rs;
}
void dfs(int u, int p){
  dpt[u] = dpt[p] + 1;
  up[u][0] = p;
  for(int i = 1; i < LG; i++)
    up[u][i] = up[up[u][i - 1]][i - 1];
  for(auto v : adj[u]){
    if(v == p)
      continue;
    dfs(v, u);
  }
}
int get_up(int u, int k){
  k--;
  if(dpt[u] <= k)
    return INF;
  for(int i = 0; i < LG; i++){
    if((1ll << i) & k)
      u = up[u][i];
  }
  return u;
}
signed main(){
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  int n;
  cin >> n;
  long long sum = 0;
  map < long long , int > mp;
  mp[0] = 0;
  for(int i = 1; i <= n; i++){
    cin >> a[i];
    sum += a[i];
    c[i] = mp.count(sum) == 0 ? -1 : mp[sum] + 1;
    mx[i] = max(mx[i - 1], c[i]);
    mp[pr[i]] = i;
  }
  build(1, 1, n);
  for(int i = 1; i <= n; i++){
    int l = i, r = n;
    int best = ask(1, 1, n, i, n, i);
    if(best == -1)
      best = n + 1;
    adj[best].push_back(i);
    adj[i].push_back(best);
  }
  dfs(n + 1, n + 1);
  int q;
  cin >> q;
  while(q--){
    int l, r;
    cin >> l >> r;
    int lst = -1, sz = 0, i = 0;
    int lo = 1, hi = r;
    while(hi >= lo){
      int mid = (lo + hi) >> 1;
      if(mx[mid] < l){
        i = mid;
        lo = mid + 1;
      }
      else hi = mid - 1;
    }
    i++;
    lo = 0, hi = INF;
    sz = (i <= r);
    while(lo <= hi){
      int mid = (lo + hi) >> 1;
      if(get_up(i, mid) <= r){
        sz = mid;
        lo = mid + 1;
      }
      else hi = mid - 1;
    }
    cout << sz << "\n";
  }
}
/*
3 2 2 2 4
3 3 3 2 4

*/

컴파일 시 표준 에러 (stderr) 메시지

sumzero.cpp: In function 'int main()':
sumzero.cpp:64:8: error: 'pr' was not declared in this scope
   64 |     mp[pr[i]] = i;
      |        ^~