1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#include <iostream>

using namespace std;

int gloskowosc[] = {1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1, 0};

int sprawdzGloskowosc(char c) {
	return gloskowosc[c - 'a'];
}

long long policzPodslowa(long long x) {
	return x*(x+1)/2 - x;
}

const int NMAX = 200000+ 5;

int litery[NMAX];

int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);

	string napis;
	long long n;

	cin >> napis;
	n = napis.length();

	for(int i = 0; i < n; i++) {
		litery[i] = sprawdzGloskowosc(napis[i]);
	}

	int ostatni = -1;
	int dlugoscSerii = 0;
	int indexOstatniegoZnalezionego = -1;
	long long temp = 0;
	long long prosteFragmenty = 0;

	for(int i = 0; i < n; i++) {
		if(litery[i] == ostatni)
			dlugoscSerii++;
		else{
			ostatni = litery[i];
			dlugoscSerii = 1;
		}

		if(dlugoscSerii >= 3) {
			temp = (i - 1) - (indexOstatniegoZnalezionego + 1) + 1;
			prosteFragmenty += policzPodslowa(temp);
			indexOstatniegoZnalezionego = i - 2;
		}
	}

	temp = (n - 1) - (indexOstatniegoZnalezionego + 1) + 1;
	prosteFragmenty += policzPodslowa(temp);

	cout << policzPodslowa(n) - prosteFragmenty;

    return 0;
}