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

#include <algorithm>     //sort,二分探索,など
#include <bitset>        //固定長bit集合
#include <cmath>         //pow,logなど
#include <complex>       //複素数
#include <deque>         //両端アクセスのキュー
#include <functional>    //sortのgreater
#include <iomanip>       //setprecision(浮動小数点の出力の誤差)
#include <iostream>      //入出力
#include <iterator>      //集合演算(積集合,和集合,差集合など)
#include <map>           //map(辞書)
#include <numeric>       //iota(整数列の生成),gcdとlcm(c++17)
#include <queue>         //キュー
#include <set>           //集合
#include <stack>         //スタック
#include <string>        //文字列
#include <unordered_map> //イテレータあるけど順序保持しないmap
#include <unordered_set> //イテレータあるけど順序保持しないset
#include <utility>       //pair
#include <vector>        //可変長配列

using namespace std;
typedef long long ll;

// マクロ
#define REP(i, n) for (ll i = 0; i < (ll)(n); i++)
#define REPD(i, n) for (ll i = (ll)(n)-1; i >= 0; i--)
#define FOR(i, a, b) for (ll i = (a); i <= (b); i++)
#define FORD(i, a, b) for (ll i = (a); i >= (b); i--)
#define ALL(x) (x).begin(), (x).end() // sortなどの引数を省略したい
#define SIZE(x) ((ll)(x).size())      // sizeをsize_tからllに直しておく
#define MAX(x) *max_element(ALL(x))
#define INF 1000000000000 // 10^12
#define MOD 1000000007    // 10^9+7
#define PB push_back
#define MP make_pair
#define F first
#define S second
#define MAXR 100000 // 10^5:最大のrange(素数列挙などで使用)

// debug
#define DEBUG(x) cout << #x << ": " << x << endl
#define DEBUG2(x, y) cout << #x << ": " << x << " " << #y << ": " << y << endl
#define DEBUG_VEC(v)                                                           \\
  cout << #v << ":";                                                           \\
  REP(i, v.size())                                                             \\
  cout << " " << v[i];                                                         \\
  cout << endl
#define DEBUG_ARR(v, n)                                                        \\
  cout << #v << ":";                                                           \\
  REP(i, n)                                                                    \\
  cout << " " << v[i];                                                         \\
  cout << endl

ll N;
vector<ll> A;
set<ll> X;
map<ll, vector<ll>> Lmap;
map<ll, vector<ll>> Rmap;

signed main() {
  ll cnt = 0;

  cin >> N;

  REP(i, N) {
    ll ai;
    cin >> ai;

    // Li = i + Ai
    ll l = i + ai;
    Lmap[l].push_back(i);
    // Rj = j - Aj
    ll r = i - ai;
    Rmap[r].push_back(i);

    // とりうるXを保存
    X.insert(l);
    X.insert(r);
  }

  for (auto x : X) {
    // cout << "x: " << x << endl;
    // DEBUG_VEC(Lmap[x]);
    // DEBUG_VEC(Rmap[x]);
    cnt += Lmap[x].size() * Rmap[x].size();
  }

  cout << cnt << endl;
}

Error: TLE

Fixed submission:

#include <algorithm>     //sort,二分探索,など
#include <bitset>        //固定長bit集合
#include <cmath>         //pow,logなど
#include <complex>       //複素数
#include <deque>         //両端アクセスのキュー
#include <functional>    //sortのgreater
#include <iomanip>       //setprecision(浮動小数点の出力の誤差)
#include <iostream>      //入出力
#include <iterator>      //集合演算(積集合,和集合,差集合など)
#include <map>           //map(辞書)
#include <numeric>       //iota(整数列の生成),gcdとlcm(c++17)
#include <queue>         //キュー
#include <set>           //集合
#include <stack>         //スタック
#include <string>        //文字列
#include <unordered_map> //イテレータあるけど順序保持しないmap
#include <unordered_set> //イテレータあるけど順序保持しないset
#include <utility>       //pair
#include <vector>        //可変長配列

using namespace std;
typedef long long ll;

// マクロ
#define REP(i, n) for (ll i = 0; i < (ll)(n); i++)
#define REPD(i, n) for (ll i = (ll)(n)-1; i >= 0; i--)
#define FOR(i, a, b) for (ll i = (a); i <= (b); i++)
#define FORD(i, a, b) for (ll i = (a); i >= (b); i--)
#define ALL(x) (x).begin(), (x).end() // sortなどの引数を省略したい
#define SIZE(x) ((ll)(x).size())      // sizeをsize_tからllに直しておく
#define MAX(x) *max_element(ALL(x))
#define INF 1000000000000 // 10^12
#define MOD 1000000007    // 10^9+7
#define PB push_back
#define MP make_pair
#define F first
#define S second
#define MAXR 100000 // 10^5:最大のrange(素数列挙などで使用)

// debug
#define DEBUG(x) cout << #x << ": " << x << endl
#define DEBUG2(x, y) cout << #x << ": " << x << " " << #y << ": " << y << endl
#define DEBUG_VEC(v)                                                           \\
  cout << #v << ":";                                                           \\
  REP(i, v.size())                                                             \\
  cout << " " << v[i];                                                         \\
  cout << endl
#define DEBUG_ARR(v, n)                                                        \\
  cout << #v << ":";                                                           \\
  REP(i, n)                                                                    \\
  cout << " " << v[i];                                                         \\
  cout << endl

ll N;
vector<ll> A;
set<ll> X;
map<ll, vector<ll>> Lmap;
map<ll, vector<ll>> Rmap;

signed main() {
  ll cnt = 0;

  cin >> N;

  REP(i, N) {
    ll ai;
    cin >> ai;

    // Li = i + Ai
    ll l = i + ai;
    Lmap[l].push_back(i);
    // Rj = j - Aj
    ll r = i - ai;
    Rmap[r].push_back(i);

    // とりうるXを保存
    X.insert(l);
    X.insert(r);
  }

  for (auto x : X) {
    // cout << "x: " << x << endl;
    // DEBUG_VEC(Lmap[x]);
    // DEBUG_VEC(Rmap[x]);
    for (auto i : Lmap[x]) {
      for (auto j : Rmap[x]) {
        if (i < j) {
          cnt++;
        }
      }
    }
  }

  cout << cnt << endl;
}