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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include <iostream>
#include <string>

using namespace std;

bool chechFor(string w, int start, int end)
{
    int underWord[3] = { 0,0,0 }; // 0 = a 1 = b 2 = c

    for (int i = start; i <= end; i++)
    {
        underWord[w[i] - 97]++;
    }


    if ((underWord[0] != 0 || underWord[1] != 0) || underWord[2] != 0)
    {
        if (underWord[0] == 0)
        {
            if (underWord[1] == 0)
                return true;
            else if (underWord[2] == 0)
                return true;
            else if (underWord[1] == underWord[2])
                return true;
        }
        else if (underWord[1] == 0)
        {
            if (underWord[0] == 0)
                return true;
            else if (underWord[2] == 0)
                return true;
            else if (underWord[0] == underWord[2])
                return true;
        }
        else if (underWord[2] == 0)
        {
            if (underWord[0] == 0)
                return true;
            else if (underWord[1] == 0)
                return true;
            else if (underWord[0] == underWord[1])
                return true;
        }
        else if (underWord[0] == underWord[1] && underWord[1] == underWord[2])
            return true;


    }

    return false;
}

int main() {
    string word = "";
    //int underWord[3] = { 0,0,0 };
    int count = 0;

    int start = 0;
    int length = 0;
    int end = 0;
    bool bal = true;

    cin >> word;

    length = word.size();
    end = length;
    count += length;
    count += length - 1;

    start = 0;
    if (length >= 3)
    {
        while (start < length - 2)
        {
            bal = true;
            end = length - 1;

            while (bal)
            {
                if (chechFor(word, start, end))
                    count++;
                end--;
                if (end - start < 2)
                    bal = false;
            }
            start++;
        }
    }

    cout << count;

    return 0;
}