Problem p02714 (C++)

Buggy submission (status: 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 <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 10000007      // 10^9+7
#define MAXR 100000       // 10^5:最大のrange(素数列挙などで使用)

int main(void) {
  int n;
  cin >> n;
  string s;
  cin >> s;
  vector<int> a, b, c;
  REP(i, n) {
    if (s[i] == 'R')
      a.push_back(i + 1);
    if (s[i] == 'G')
      b.push_back(i + 1);
    if (s[i] == 'B')
      c.push_back(i + 1);
  }
  ll sum = 0;
  // cout << a.size() << "," << b.size() << "," << c.size() << endl;
  REP(i, a.size()) {
    REP(j, b.size()) {
      REP(k, c.size()) {
        if (a[i] < b[j]) {
          if (b[j] < c[k]) { // c, b, a
            if (c[k] - b[j] == b[j] - a[i])
              continue;
          }
          if (c[k] < a[i]) { // b, a, c
            if (b[j] - a[i] == a[i] - c[k])
              continue;
          }
          if (b[j] > c[k] && c[k] > a[i]) { // b, c, a
            if (b[j] - c[k] == c[k] - a[i])
              continue;
          }
        } else {
          if (a[i] < c[k]) { // c, a, b
            if (c[k] - a[i] == a[i] - b[j])
              continue;
          }
          if (b[j] > c[k]) { // a, b, c
            if (a[i] - b[j] == b[j] - c[k])
              continue;
          }
          if (a[i] > c[k] && c[k] > b[j]) { // a, c, b
            if (a[i] - c[k] == c[k] - b[j])
              continue;
          }
        }
        sum++;
      }
    }
  }

  cout << sum << endl;
  return 0;
}


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 <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 10000007      // 10^9+7
#define MAXR 100000       // 10^5:最大のrange(素数列挙などで使用)

int main(void) {
  int n;
  cin >> n;
  string s;
  cin >> s;
  vector<int> a, b, c;
  REP(i, n) {
    if (s[i] == 'R')
      a.push_back(i + 1);
    if (s[i] == 'G')
      b.push_back(i + 1);
    if (s[i] == 'B')
      c.push_back(i + 1);
  }
  ll sum = a.size() * b.size() * c.size();

  // とびとびのものを数えて引く
  for (int i = 1; i < n; i++) {
    for (int j = 0; j + i + i < n; j++) {
      if (s[j] == s[j + i] || s[j + i] == s[j + i + i] || s[j] == s[j + i + i])
        continue;
      sum--;
    }
  }

  cout << sum << endl;
  return 0;
}