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
62
#include <bits/stdc++.h>
using namespace std;

int	ile_a_pref[300010];
int ile_b_pref[300010];
int ile_c_pref[300010];


int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int ile_a, ile_b, ile_c;
	int odp=0;
	int n;
	string pom;
	string slowo;
	cin>>slowo;
	n=slowo.size();
	slowo="/"+slowo;
	for(int i=1; i<=n; i++)
	{
		ile_a_pref[i]=ile_a_pref[i-1];
		ile_b_pref[i]=ile_b_pref[i-1];
		ile_c_pref[i]=ile_c_pref[i-1];
		switch(slowo[i])
		{
			case 'a': ile_a_pref[i]=ile_a_pref[i-1]+1;
			break;
			case 'b': ile_b_pref[i]=ile_b_pref[i-1]+1;
			break;
			case 'c': ile_c_pref[i]=ile_c_pref[i-1]+1;
		}
	}
	

	
	for(int i=1; i<=n; i++)
	{
		for(int j=i; j<=n; j++)
		{
			ile_a=ile_a_pref[j]-ile_a_pref[i-1];
			ile_b=ile_b_pref[j]-ile_b_pref[i-1];
			ile_c=ile_c_pref[j]-ile_c_pref[i-1];	
		
			if((ile_a==ile_b && ile_b==ile_c) ||
			 (ile_a==ile_b && ile_c==0) ||
			 (ile_a==ile_c && ile_b==0) ||
			 (ile_c==ile_b && ile_a==0) ||
			 (ile_c>0 && ile_a==0 && ile_b==0) ||
			 (ile_b>0 && ile_a==0 && ile_c==0) ||
			 (ile_a>0 && ile_c==0 && ile_b==0))
			{
				odp++;
			
			}
		}
	}
	cout<<odp<<endl;
	return 0;
}