#include <iostream>
#include <string>
using namespace std;
#define FORD(x, b, e) for (int x = b; x >= (e); --x)
#define REP(x, n) for (int x = 0; x < (n); ++x)
#define SIZE(x) ((int)(x).size())
int a, b, c;
int n;
int ans = 0;
string word;
int main()
{
cin >> word;
n = SIZE(word);
REP(i, n)
{
a = b = c = 0;
FORD(j, i, 0)
{
if (word[j] == 'a') a++;
else if (word[j] == 'b') b++;
else if (word[j] == 'c') c++;
// if a == 0 -> only b and c are important
if (a == 0)
{
if (b == 0)
ans++;
else if (c == 0)
ans++;
else if (b == c)
ans++;
}
// if b == 0 and a != 0
else if (b == 0)
{
if (c == 0)
ans++;
else if (a == c)
ans++;
}
// if c == 0 and a != 0 and b != 0
else if (c == 0)
{
if (a == b)
ans++;
}
// no letter is equal to zero
else if (a == b && b == c)
{
ans++;
}
}
}
cout << ans;
}
/*
aabbabcccba
*/
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 | #include <iostream> #include <string> using namespace std; #define FORD(x, b, e) for (int x = b; x >= (e); --x) #define REP(x, n) for (int x = 0; x < (n); ++x) #define SIZE(x) ((int)(x).size()) int a, b, c; int n; int ans = 0; string word; int main() { cin >> word; n = SIZE(word); REP(i, n) { a = b = c = 0; FORD(j, i, 0) { if (word[j] == 'a') a++; else if (word[j] == 'b') b++; else if (word[j] == 'c') c++; // if a == 0 -> only b and c are important if (a == 0) { if (b == 0) ans++; else if (c == 0) ans++; else if (b == c) ans++; } // if b == 0 and a != 0 else if (b == 0) { if (c == 0) ans++; else if (a == c) ans++; } // if c == 0 and a != 0 and b != 0 else if (c == 0) { if (a == b) ans++; } // no letter is equal to zero else if (a == b && b == c) { ans++; } } } cout << ans; } /* aabbabcccba */ |
English