#aBC162D. [ABC162D] RGB Triplets

[ABC162D] RGB Triplets

AT_abc162_d [ABC162D] RGB Triplets

题目描述

有一个只由 RGB 组成的长度为 NN 的字符串 SS

请计算满足以下两个条件的三元组 (i, j, k)(i,~j,~k) 的数量(1i<j<kN1 \leq i < j < k \leq N):

  • SiSjS_i \neq S_j,且 SiSkS_i \neq S_k,且 SjSkS_j \neq S_k
  • jikjj - i \neq k - j

输入格式

输入以以下格式从标准输入读入。

NN SS

输出格式

输出满足题意的三元组数量。

输入输出样例 #1

输入 #1

4
RRGB

输出 #1

1

输入输出样例 #2

输入 #2

39
RBRBGRBGGBBRRGBBRRRBGGBRBGBRBGBRBBBGBBB

输出 #2

1800

说明/提示

限制

  • 1N40001 \leq N \leq 4000
  • SS 是一个只包含 RGB 的长度为 NN 的字符串

样例解释 1

只有三元组 (1, 3, 4)(1,~3,~4) 同时满足两个条件。三元组 (2, 3, 4)(2,~3,~4) 虽然满足第一个条件,但不满足第二个条件,因此不计入答案。

由 ChatGPT 4.1 翻译