Problem p02537 (C++) — buggy submission, verdict: Runtime Error

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

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

const int INF = 2147483647;

/* RMQ：[0,n-1] について、区間ごとの最小値を管理する構造体
    set(int i, T x), build():
   i番目の要素をxにセット。まとめてセグ木を構築する。O(n) update(i,x): i
   番目の要素を x に更新。O(log(n)) query(a,b): [a,b)
   での最小の要素を取得。O(log(n)) find_rightest(a,b,x): [a,b) で
   x以下の要素を持つ最右位置を求める。O(log(n)) find_leftest(a,b,x): [a,b) で
   x以下の要素を持つ最左位置を求める。O(log(n))
*/
template <typename X> struct SegTree {
  using FX = function<X(X, X)>; // X•X -> X となる関数の型
  int n;
  FX fx;
  const X ex;
  vector<X> dat;
  SegTree(int n_, FX fx_, X ex_) : n(), fx(fx_), ex(ex_), dat(n_ * 4, ex_) {
    int x = 1;
    while (n_ > x) {
      x *= 2;
    }
    n = x;
  }
  void set(int i, X x) { dat[i + n - 1] = x; }
  void build() {
    for (int k = n - 2; k >= 0; k--)
      dat[k] = fx(dat[2 * k + 1], dat[2 * k + 2]);
  }
  void update(int i, X x) {
    i += n - 1;
    dat[i] = x;
    while (i > 0) {
      i = (i - 1) / 2; // parent
      dat[i] = fx(dat[i * 2 + 1], dat[i * 2 + 2]);
    }
  }
  X get(int i) { return dat[i + n - 1]; }
  X all() { return dat[0]; }
  X query(int a, int b) { return query_sub(a, b, 0, 0, n); }
  X query_sub(int a, int b, int k, int l, int r) {
    if (r <= a || b <= l) {
      return ex;
    } else if (a <= l && r <= b) {
      return dat[k];
    } else {
      X vl = query_sub(a, b, k * 2 + 1, l, (l + r) / 2);
      X vr = query_sub(a, b, k * 2 + 2, (l + r) / 2, r);
      return fx(vl, vr);
    }
  }
  int find_rightest(int a, int b, X x) {
    return find_rightest_sub(a, b, x, 0, 0, n);
  }
  int find_leftest(int a, int b, X x) {
    return find_leftest_sub(a, b, x, 0, 0, n);
  }
  int find_rightest_sub(int a, int b, X x, int k, int l, int r) {
    if (dat[k] > x || r <= a ||
        b <= l) { // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn a-1
      return a - 1;
    } else if (k >= n - 1) { // 自分が葉ならその位置をreturn
      return (k - (n - 1));
    } else {
      int vr = find_rightest_sub(a, b, x, 2 * k + 2, (l + r) / 2, r);
      if (vr != a - 1) { // 右の部分木を見て a-1 以外ならreturn
        return vr;
      } else { // 左の部分木を見て値をreturn
        return find_rightest_sub(a, b, x, 2 * k + 1, l, (l + r) / 2);
      }
    }
  }
  int find_leftest_sub(int a, int b, X x, int k, int l, int r) {
    if (dat[k] > x || r <= a ||
        b <= l) { // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn b
      return b;
    } else if (k >= n - 1) { // 自分が葉ならその位置をreturn
      return (k - (n - 1));
    } else {
      int vl = find_leftest_sub(a, b, x, 2 * k + 1, l, (l + r) / 2);
      if (vl != b) { // 左の部分木を見て b 以外ならreturn
        return vl;
      } else { // 右の部分木を見て値をreturn
        return find_leftest_sub(a, b, x, 2 * k + 2, (l + r) / 2, r);
      }
    }
  }
};

int main() {
  int n, k;
  cin >> n >> k;

  vector<int> A(n);
  rep(i, n) cin >> A[i];

  // fx:二項演算,ex:単位元,0-index
  using X = int;
  auto fx = [](X x1, X x2) -> X { return max(x1, x2); };
  X ex = 0;
  SegTree<X> seg(300010, fx, ex);

  rep(i, n) {
    int l = max(0, A[i] - k);
    int r = min(300000, A[i] + k);
    int tmp = seg.query(l, r + 1) + 1;
    seg.update(A[i], tmp);
  }
  cout << seg.all() << endl;

  return 0;
}

Error: 0

Fixed submission:

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

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

const int INF = 2147483647;

/* RMQ：[0,n-1] について、区間ごとの最小値を管理する構造体
    set(int i, T x), build():
   i番目の要素をxにセット。まとめてセグ木を構築する。O(n) update(i,x): i
   番目の要素を x に更新。O(log(n)) query(a,b): [a,b)
   での最小の要素を取得。O(log(n)) find_rightest(a,b,x): [a,b) で
   x以下の要素を持つ最右位置を求める。O(log(n)) find_leftest(a,b,x): [a,b) で
   x以下の要素を持つ最左位置を求める。O(log(n))
*/
template <typename X> struct SegTree {
  using FX = function<X(X, X)>; // X•X -> X となる関数の型
  int n;
  FX fx;
  const X ex;
  vector<X> dat;
  SegTree(int n_, FX fx_, X ex_) : n(), fx(fx_), ex(ex_), dat(n_ * 4, ex_) {
    int x = 1;
    while (n_ > x) {
      x *= 2;
    }
    n = x;
  }
  void set(int i, X x) { dat[i + n - 1] = x; }
  void build() {
    for (int k = n - 2; k >= 0; k--)
      dat[k] = fx(dat[2 * k + 1], dat[2 * k + 2]);
  }
  void update(int i, X x) {
    i += n - 1;
    dat[i] = x;
    while (i > 0) {
      i = (i - 1) / 2; // parent
      dat[i] = fx(dat[i * 2 + 1], dat[i * 2 + 2]);
    }
  }
  X get(int i) { return dat[i + n - 1]; }
  X all() { return dat[1]; }
  X query(int a, int b) { return query_sub(a, b, 0, 0, n); }
  X query_sub(int a, int b, int k, int l, int r) {
    if (r <= a || b <= l) {
      return ex;
    } else if (a <= l && r <= b) {
      return dat[k];
    } else {
      X vl = query_sub(a, b, k * 2 + 1, l, (l + r) / 2);
      X vr = query_sub(a, b, k * 2 + 2, (l + r) / 2, r);
      return fx(vl, vr);
    }
  }
  int find_rightest(int a, int b, X x) {
    return find_rightest_sub(a, b, x, 0, 0, n);
  }
  int find_leftest(int a, int b, X x) {
    return find_leftest_sub(a, b, x, 0, 0, n);
  }
  int find_rightest_sub(int a, int b, X x, int k, int l, int r) {
    if (dat[k] > x || r <= a ||
        b <= l) { // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn a-1
      return a - 1;
    } else if (k >= n - 1) { // 自分が葉ならその位置をreturn
      return (k - (n - 1));
    } else {
      int vr = find_rightest_sub(a, b, x, 2 * k + 2, (l + r) / 2, r);
      if (vr != a - 1) { // 右の部分木を見て a-1 以外ならreturn
        return vr;
      } else { // 左の部分木を見て値をreturn
        return find_rightest_sub(a, b, x, 2 * k + 1, l, (l + r) / 2);
      }
    }
  }
  int find_leftest_sub(int a, int b, X x, int k, int l, int r) {
    if (dat[k] > x || r <= a ||
        b <= l) { // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn b
      return b;
    } else if (k >= n - 1) { // 自分が葉ならその位置をreturn
      return (k - (n - 1));
    } else {
      int vl = find_leftest_sub(a, b, x, 2 * k + 1, l, (l + r) / 2);
      if (vl != b) { // 左の部分木を見て b 以外ならreturn
        return vl;
      } else { // 右の部分木を見て値をreturn
        return find_leftest_sub(a, b, x, 2 * k + 2, (l + r) / 2, r);
      }
    }
  }
};

int main() {
  int n, k;
  cin >> n >> k;

  vector<int> A(n);
  rep(i, n) cin >> A[i];

  // fx:二項演算,ex:単位元,0-index
  using X = int;
  auto fx = [](X x1, X x2) -> X { return max(x1, x2); };
  X ex = 0;
  SegTree<X> seg(n, fx, ex);

  rep(i, n) {
    int l = max(0, A[i] - k);
    int r = min(n - 1, A[i] + k);
    int tmp = seg.query(l, r + 1) + 1;
    seg.update(A[i], tmp);
  }
  cout << seg.all() << endl;

  return 0;
}