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

#define _GLIBCXX_DEBUG
#include "bits/stdc++.h"
// #include <intrin.h>  //AtCoder (gcc)
// 上ではこれがあると動かない。__popcnt用のincludeファイル。
using namespace std;

typedef long long ll;
typedef long double ld;

#define int long long
#define rep(i, n) for (long long i = 0; i < (n); i++)
#define sqrt(d) pow((long double)(d), 0.50)

const int INF = 2000000000;     // 2e9
const int large_P = 1000000007; // 1e9 + 7

// 繰り返し2乗法
// N^aの、Mで割った余りを求める。
ll my_pow(ll N, ll a, ll M) {
  ll tempo;
  if (a == 0) {
    return 1;
  } else {
    if (a % 2 == 0) {
      tempo = my_pow(N, a / 2, M);
      return (tempo * tempo) % M;
    } else {
      tempo = my_pow(N, a - 1, M);
      return (tempo * N) % M;
    }
  }
}

// N_C_a を M で割った余り
ll my_combination(ll N, ll a, ll M) {
  ll answer = 1;

  rep(i, a) {
    answer *= (N - i);
    answer %= M;
  }

  rep(i, a) {
    answer *= my_pow(i + 1, M - 2, M);
    answer %= M;
  }

  return answer;
}

ll my_gcd(ll &a, ll &b) {
  if (b == 0)
    return a;
  ll temp = a % b;
  return my_gcd(b, temp);
}

// base を底としたときの、n の i桁目を、v.at(i) に入れる。(桁数は n
// に応じて自動で設定する。)
void ll_to_vector(int base, long long n, vector<int> &v) {
  long long tempo = n;
  long long tempo2 = n;

  // ll n_digit = floor(log10(n) / log10(base)) + 1;  //
  // 桁の計算を1行で。ただ誤差があるのでよくない。
  int n_digit = 1;
  while (tempo2 >= 10) {
    tempo2 /= 10;
    n_digit++;
  }

  v.assign(n_digit, 0);

  // ll n_digit = v.size();  // v のサイズをそのままにする場合。

  for (int i = 0; i < n_digit; i++) {
    v.at(i) = tempo / pow(base, n_digit - 1 - i);
    tempo -= v.at(i) * pow(base, n_digit - 1 - i);
  }
}

int char_to_int(char c) {
  switch (c) {
  case '0':
    return 0;
  case '1':
    return 1;
  case '2':
    return 2;
  case '3':
    return 3;
  case '4':
    return 4;
  case '5':
    return 5;
  case '6':
    return 6;
  case '7':
    return 7;
  case '8':
    return 8;
  case '9':
    return 9;
  default:
    return 0;
  }
}

// エラトステネスの篩で、prime で ないところに false を入れる。
// true で初期化された vector を代入する必要がある。
void prime_judge(vector<bool> &prime_or_not) {

  prime_or_not.at(0) = false;
  prime_or_not.at(1) = false;

  ll n = prime_or_not.size() - 1;
  ll region = (ll)sqrt(n) + 1;

  for (ll i = 2; i <= n / 2; i++) {
    prime_or_not.at(2 * i) = false;
  }

  for (ll i2 = 1; i2 < n / 2; i2++) {
    ll i = 2 * i2 + 1; // ここからは奇数のみ探索
    if (i < region && prime_or_not.at(i)) {
      ll j = i * i;
      while (j < n + 1) {
        prime_or_not.at(j) = false;
        j += 2 * i;
      }
    }
  }
};

map<ll, ll> divide_to_prime(int target) {
  map<ll, ll> res;

  // sqrt(target) まで調べる。
  ll upper_lim = ceil(sqrt(target));
  vector<bool> prime_or_not(upper_lim + 1, true);

  if (upper_lim < 20)
    prime_or_not.assign(25, true);
  prime_judge(prime_or_not);

  ll tempo = target;
  rep(i, (upper_lim + 1)) {
    if (prime_or_not.at(i)) {
      while (tempo % i == 0) {
        tempo /= i;
        res[i]++;
      }
    }
  }

  if (tempo != 1)
    res[tempo]++; // sqrt(target) より大きな素因数は高々1つしかない。

  return res;
}

signed main() {
  int N;
  cin >> N;
  vector<int> A(N);
  rep(i, N) cin >> A.at(i);

  vector<int> left(N), right(N);
  map<int, int> memo;

  rep(i, N) {
    left.at(i) = i + A.at(i);
    right.at(i) = i - A.at(i);
    memo[i - A.at(i)]++;
  }

  int res = 0;
  rep(i, N) {
    memo[left.at(i)]++;
    memo[left.at(i)]--;
    res += memo.at(left.at(i));
  }

  cout << res << endl;
}

Error: TLE

Fixed submission:

// #define _GLIBCXX_DEBUG
#include "bits/stdc++.h"
// #include <intrin.h>  //AtCoder (gcc)
// 上ではこれがあると動かない。__popcnt用のincludeファイル。
using namespace std;

typedef long long ll;
typedef long double ld;

#define int long long
#define rep(i, n) for (long long i = 0; i < (n); i++)
#define sqrt(d) pow((long double)(d), 0.50)

const int INF = 2000000000;     // 2e9
const int large_P = 1000000007; // 1e9 + 7

// 繰り返し2乗法
// N^aの、Mで割った余りを求める。
ll my_pow(ll N, ll a, ll M) {
  ll tempo;
  if (a == 0) {
    return 1;
  } else {
    if (a % 2 == 0) {
      tempo = my_pow(N, a / 2, M);
      return (tempo * tempo) % M;
    } else {
      tempo = my_pow(N, a - 1, M);
      return (tempo * N) % M;
    }
  }
}

// N_C_a を M で割った余り
ll my_combination(ll N, ll a, ll M) {
  ll answer = 1;

  rep(i, a) {
    answer *= (N - i);
    answer %= M;
  }

  rep(i, a) {
    answer *= my_pow(i + 1, M - 2, M);
    answer %= M;
  }

  return answer;
}

ll my_gcd(ll &a, ll &b) {
  if (b == 0)
    return a;
  ll temp = a % b;
  return my_gcd(b, temp);
}

// base を底としたときの、n の i桁目を、v.at(i) に入れる。(桁数は n
// に応じて自動で設定する。)
void ll_to_vector(int base, long long n, vector<int> &v) {
  long long tempo = n;
  long long tempo2 = n;

  // ll n_digit = floor(log10(n) / log10(base)) + 1;  //
  // 桁の計算を1行で。ただ誤差があるのでよくない。
  int n_digit = 1;
  while (tempo2 >= 10) {
    tempo2 /= 10;
    n_digit++;
  }

  v.assign(n_digit, 0);

  // ll n_digit = v.size();  // v のサイズをそのままにする場合。

  for (int i = 0; i < n_digit; i++) {
    v.at(i) = tempo / pow(base, n_digit - 1 - i);
    tempo -= v.at(i) * pow(base, n_digit - 1 - i);
  }
}

int char_to_int(char c) {
  switch (c) {
  case '0':
    return 0;
  case '1':
    return 1;
  case '2':
    return 2;
  case '3':
    return 3;
  case '4':
    return 4;
  case '5':
    return 5;
  case '6':
    return 6;
  case '7':
    return 7;
  case '8':
    return 8;
  case '9':
    return 9;
  default:
    return 0;
  }
}

// エラトステネスの篩で、prime で ないところに false を入れる。
// true で初期化された vector を代入する必要がある。
void prime_judge(vector<bool> &prime_or_not) {

  prime_or_not.at(0) = false;
  prime_or_not.at(1) = false;

  ll n = prime_or_not.size() - 1;
  ll region = (ll)sqrt(n) + 1;

  for (ll i = 2; i <= n / 2; i++) {
    prime_or_not.at(2 * i) = false;
  }

  for (ll i2 = 1; i2 < n / 2; i2++) {
    ll i = 2 * i2 + 1; // ここからは奇数のみ探索
    if (i < region && prime_or_not.at(i)) {
      ll j = i * i;
      while (j < n + 1) {
        prime_or_not.at(j) = false;
        j += 2 * i;
      }
    }
  }
};

map<ll, ll> divide_to_prime(int target) {
  map<ll, ll> res;

  // sqrt(target) まで調べる。
  ll upper_lim = ceil(sqrt(target));
  vector<bool> prime_or_not(upper_lim + 1, true);

  if (upper_lim < 20)
    prime_or_not.assign(25, true);
  prime_judge(prime_or_not);

  ll tempo = target;
  rep(i, (upper_lim + 1)) {
    if (prime_or_not.at(i)) {
      while (tempo % i == 0) {
        tempo /= i;
        res[i]++;
      }
    }
  }

  if (tempo != 1)
    res[tempo]++; // sqrt(target) より大きな素因数は高々1つしかない。

  return res;
}

signed main() {
  int N;
  cin >> N;
  vector<int> A(N);
  rep(i, N) cin >> A.at(i);

  int res = 0;
  rep(i, N) {
    if (i + A.at(i) + 1 >= N)
      continue;
    else {
      for (int j = 0; j < N - (i + A.at(i) + 1); j++) {

        if (j + 1 == A.at(j + i + A.at(i) + 1))
          res++;
      }
    }
  }

  cout << res << endl;
}