提出 #11935231

---

ソースコード 拡げる

#include <bits/stdc++.h>
#define INF 1e9
using namespace std;

#define REPR(i,n) for(int i=(n); i >= 0; --i)
#define FOR(i, m, n) for(int i = (m); i < (n); ++i)
#define REP(i, n) for(int i=0, i##_len=(n); i<i##_len; ++i)
#define ALL(a)  (a).begin(),(a).end()
#define endl "\n"

template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return true; } return false; }
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return true; } return false; }
typedef long long ll;

string S;
int N;
// pos
vector<set<int>> v;

// j - i \ne k-j を満たす数を負数として返す。
ll go() {
    ll ans = 0;
    for(const auto &i:v[0]) {
        // find start
        auto sj = upper_bound(ALL(v[1]),i);
        for(auto j = sj; j != v[1].end(); j ++) {
            int nk = 2*(*j) - i;
            if (   *j < nk && v[2].count(nk) == 1)  ans--;
        }
    }
    return ans;
}



void solve() {
    v.resize(3);
    cin >> N;
    cin >> S;
    REP(i,N) {
        if (S[i]== 'R') v[0].insert(i);
        if (S[i]=='G') v[1].insert(i);
        if (S[i]=='B') v[2].insert(i);
    }
    ll ans = v[0].size()*v[1].size()*v[2].size();
    // iter_swapでi,j,kとR,G,Bの袋の割り当て方を全部だしている。
    REP(i,3) {
        ans += go();
        iter_swap(v.begin()+1,v.begin()+2);
        ans += go();
        iter_swap(v.begin(),v.begin()+1);
    }
    cout << ans << endl;

}

int main() {
    solve();
    return 0;
}

提出情報

ジャッジ結果