Problem p02537 (C++) — buggy submission, verdict: Time Limit Exceeded

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
                  　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <vector>

////多倍長整数, cpp_intで宣言
// #include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;
//
#pragma GCC target("avx2")
#pragma GCC optimization("O3")
#pragma GCC optimization("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define repeat(i, n, m) for (int i = n; i < (m); ++i)
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\\n" : "no\\n")
#define printyn(a) printf(a ? "Yes\\n" : "No\\n")
#define printYN(a) printf(a ? "YES\\n" : "NO\\n")
#define printim(a) printf(a ? "possible\\n" : "imposible\\n")
#define printdb(a) printf("%.50lf\\n", a)    // 少数出力
#define printLdb(a) printf("%.50Lf\\n", a)   // 少数出力
#define printdbd(a) printf("%.16lf\\n", a)   // 少数出力(桁少なめ)
#define prints(s) printf("%s\\n", s.c_str()) // string出力
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg) / 360.0L) * 2.0L * PI)
#define rad_to_deg(rad) (((rad) / 2.0L / PI) * 360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d)                                             \\
  (abs(a - c) + abs(b - d)) /*(a, b) から (c, d) のマンハッタン距離 */
#define inf numeric_limits<double>::infinity();
#define linf numeric_limits<long double>::infinity()

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string &str) {
  char c;
  str = "";
  scanf("%c", &c);
  if (c == '\\n')
    scanf("%c", &c);
  while (c != '\\n' && c != -1 && c != ' ') {
    str += c;
    scanf("%c", &c);
  }
}

void scanc(char &str) {
  char c;
  scanf("%c", &c);
  if (c == -1)
    return;
  while (c == '\\n') {
    scanf("%c", &c);
  }
  str = c;
}

double acot(double x) { return PI / 2 - atan(x); }

ll LSB(ll n) { return (n & (-n)); }

template <typename T> inline T chmin(T &a, const T &b) {
  if (a > b)
    a = b;
  return a;
}

template <typename T> inline T chmax(T &a, const T &b) {
  if (a < b)
    a = b;
  return a;
}

////atcoder library
// #include <atcoder/all>
// using namespace atcoder;

/*-----------------------------------------ここからコード-----------------------------------------*/

/*
 * @title segment-tree
 * @docs kyopro/docs/segtree.md
 */

// セグ木/0-indexed/非再帰/(大きさ, 単位元)で初期化
template <typename T, typename F> struct segtree {
  // 木を配列であらわしたもの
  vector<T> seg;
  // 木の1/2の大きさ
  int siz;
  // 単位元
  const T e;
  ////比較関数の型
  // using F = function<T(T, T)>;
  // マージする関数
  const F f;

  // n の大きさ, a (単位元) で segtree を初期化する
  segtree(int n, const T a, const F f) : e(a), f(f) {
    siz = 1;
    while (siz < n)
      siz <<= 1;
    seg.assign(2 * siz - 1, e);
    --siz;
  }

  // k (0-indexed) 番目に t を代入
  void set(int k, const T &t) { seg[k + siz] = t; }

  // f によって木を構築
  void build() {
    for (int i = siz - 1; i >= 0; --i)
      seg[i] = f(seg[i * 2 + 1], seg[i * 2 + 2]);
  }

  // i 番目の要素を返す
  T operator[](const int i) { return seg[i + siz]; }

  // k 番目の値を a に更新
  void update(int k, T a) {
    k += siz;
    // 必要であればここを変える
    seg[k] = a;
    while (k > 0) {
      k = ((k - 1) >> 1);
      seg[k] = f(seg[k * 2 + 1], seg[k * 2 + 2]);
    }
  }

  //[a, b) について f した結果を返す
  T query(int a, int b) {
    T l = e, r = e;
    for (a += siz, b += siz; a < b; a >>= 1, b >>= 1) {
      if (!(a & 1))
        l = f(l, seg[a++]);
      if (!(b & 1))
        r = f(seg[--b], r);
    }
    return f(l, r);
  }

  //[start, end) について、[l, r) を調べながら k 番目が check を満たすか二分探索
  //最後が true なら left, false なら right fの逆演算
  template <typename C, typename FT>
  int find(const int start, const int end, int l, int r, int k, const C check,
           T &checknum, const bool b, const FT revf) {
    // cerr << checknum << '\\n';
    // 範囲外またはそこがすでに満たさないとき
    // cerr << k << ',' << checknum << '\\n';
    if (start <= l && r <= end && !check(seg[k], checknum)) {
      checknum = revf(checknum, seg[k]);
      return -1;
    }
    if ((r <= start || l >= end)) {
      return -1;
    }
    // 既に葉
    if (k >= siz) {
      return k - siz;
    }
    int res;
    if (b) {
      // 左側を調べる
      res = find<C, FT>(start, end, l, ((l + r) >> 1), (k << 1) + 1, check,
                        checknum, b, revf);
      // 左側が適してたらそれが答え
      if (res != -1)
        return (res);
      return find<C, FT>(start, end, ((l + r) >> 1), r, (k << 1) + 2, check,
                         checknum, b, revf);
    } else {
      // 右側を調べる
      res = find<C, FT>(start, end, ((l + r) >> 1), r, (k << 1) + 2, check,
                        checknum, b, revf);
      // 右側が適してたらそれが答え
      if (res != -1)
        return (res);
      return find<C, FT>(start, end, l, ((l + r) >> 1), (k << 1) + 1, check,
                         checknum, b, revf);
    }
  }

  template <typename C, typename FT>
  int find_left(int start, int end, const C check, T checknum, FT revf) {
    return find<C, FT>(start, end, 0, siz + 1, 0, check, checknum, true, revf);
  }

  template <typename C, typename FT>
  int find_right(int start, int end, const C check, T checknum, FT revf) {
    return find<C, FT>(start, end, 0, siz + 1, 0, check, checknum, false, revf);
  }
};

int main() {

  int n, k;
  scanf("%d%d", &n, &k);
  auto f = [](int a, int b) { return max(a, b); };
  segtree<int, decltype(f)> seg(300001, 0, f);
  seg.build();
  int ans = 1, a;
  rep(i, n) {
    scanf("%d", &a);
    int tmp = seg.query(max(0, a - k), min(a + k + 1, 300001)) + 1;
    chmax(ans, tmp);
    seg.update(a, tmp);
  }
  printf("%d\\n", ans);

  Please AC;
}

Error: TLE

Fixed submission:

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
                  　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstring>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <vector>

////多倍長整数, cpp_intで宣言
// #include <boost/multiprecision/cpp_int.hpp>
// using namespace boost::multiprecision;
//
// #pragma GCC target ("avx2")
// #pragma GCC optimization ("Ofast")
// #pragma GCC optimization ("unroll-loops")
// #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define repeat(i, n, m) for (int i = n; i < (m); ++i)
#define rep(i, n) for (int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\\n" : "no\\n")
#define printyn(a) printf(a ? "Yes\\n" : "No\\n")
#define printYN(a) printf(a ? "YES\\n" : "NO\\n")
#define printim(a) printf(a ? "possible\\n" : "imposible\\n")
#define printdb(a) printf("%.50lf\\n", a)    // 少数出力
#define printLdb(a) printf("%.50Lf\\n", a)   // 少数出力
#define printdbd(a) printf("%.16lf\\n", a)   // 少数出力(桁少なめ)
#define prints(s) printf("%s\\n", s.c_str()) // string出力
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg) / 360.0L) * 2.0L * PI)
#define rad_to_deg(rad) (((rad) / 2.0L / PI) * 360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d)                                             \\
  (abs(a - c) + abs(b - d)) /*(a, b) から (c, d) のマンハッタン距離 */
#define inf numeric_limits<double>::infinity();
#define linf numeric_limits<long double>::infinity()

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string &str) {
  char c;
  str = "";
  scanf("%c", &c);
  if (c == '\\n')
    scanf("%c", &c);
  while (c != '\\n' && c != -1 && c != ' ') {
    str += c;
    scanf("%c", &c);
  }
}

void scanc(char &str) {
  char c;
  scanf("%c", &c);
  if (c == -1)
    return;
  while (c == '\\n') {
    scanf("%c", &c);
  }
  str = c;
}

double acot(double x) { return PI / 2 - atan(x); }

ll LSB(ll n) { return (n & (-n)); }

template <typename T> inline T chmin(T &a, const T &b) {
  if (a > b)
    a = b;
  return a;
}

template <typename T> inline T chmax(T &a, const T &b) {
  if (a < b)
    a = b;
  return a;
}

////atcoder library
// #include <atcoder/all>
// using namespace atcoder;

/*-----------------------------------------ここからコード-----------------------------------------*/

int main() {

  int n, k;
  scanf("%d%d", &n, &k);
  vector<int> a(n);
  rep(i, n) { scanf("%d", &a[i]); }
  vector<int> dp(n);
  fill(all(dp), -1);
  dp[0] = 0;
  int ans = 0;
  rep(i, n) {
    for (int j = i + 1; j < n; ++j) {
      if (dp[j] > dp[i] + 1)
        break;
      if (abs(a[j] - a[i]) <= k)
        chmax(dp[j], dp[i] + 1);
    }
  }
  printf("%d\\n", dp[n - 1] + 1);

  Please AC;
}