题库练习 Good Triple
← 上一题 下一题 →

A12611 | Good Triple

时间限制1s
内存限制256MB
通过 / 提交0/0

题目描述

Toad Rash has a binary string $s$ . A binary string consists only of zeros and ones.

Let $n$ be the length of $s$ .

Rash needs to find the number of such pairs of integers $l$ , $r$ that $1 \leq l \leq r \leq n$ and there is at least one pair of integers $x$ , $k$ such that $1 \leq x, k \leq n$ , $l \leq x < x + 2k \leq r$ , and $s_x = s_{x+k} = s_{x+2k}$ .

Find this number of pairs for Rash.

输入格式

The first line contains the string $s$ ( $1 \leq |s| \leq 300\,000$ ), consisting of zeros and ones.

输出格式

Output one integer: the number of such pairs of integers $l$ , $r$ that $1 \leq l \leq r \leq n$ and there is at least one pair of integers $x$ , $k$ such that $1 \leq x, k \leq n$ , $l \leq x < x + 2k \leq r$ , and $s_x = s_{x+k} = s_{x+2k}$ .

输入输出样例

输入 #1
010101
输出 #1
3
输入 #2
11001100
输出 #2
0
C++ 编辑器
输入
输出