#include <iostream>
#include <vector>
#include <string>
using namespace std;
using ull = unsigned long long;
const int maxN = 200000;
bool alphabet[] = { true, false, false, false, true, false, false, false, true, false, false, false, false, false, true,
false, false, false, false, false, true, false, false, false, true, false };
ull findAllTriples(const string& str)
{
ull result = 0;
ull includedLeft = 0;
for (int i = 0; i < static_cast<int>(str.size()) - 2; ++i)
{
const bool firstIsVowel = alphabet[str[i] - 'a'];
const bool secondIsVowel = alphabet[str[i + 1] - 'a'];
const bool thirdIsVowel = alphabet[str[i + 2] - 'a'];
if ((firstIsVowel && secondIsVowel && thirdIsVowel) || (!firstIsVowel && !secondIsVowel && !thirdIsVowel))
{
result += (i + 1 - includedLeft) * (str.size() - i - 3 + 1);
includedLeft = i + 1;
}
}
return result;
}
int main()
{
ios::sync_with_stdio(false);
string str;
str.reserve(maxN);
cin >> str;
if (str.size() < 3)
{
cout << 0;
}
else
{
cout << findAllTriples(str);
}
cout << endl;
}
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 | #include <iostream> #include <vector> #include <string> using namespace std; using ull = unsigned long long; const int maxN = 200000; bool alphabet[] = { true, false, false, false, true, false, false, false, true, false, false, false, false, false, true, false, false, false, false, false, true, false, false, false, true, false }; ull findAllTriples(const string& str) { ull result = 0; ull includedLeft = 0; for (int i = 0; i < static_cast<int>(str.size()) - 2; ++i) { const bool firstIsVowel = alphabet[str[i] - 'a']; const bool secondIsVowel = alphabet[str[i + 1] - 'a']; const bool thirdIsVowel = alphabet[str[i + 2] - 'a']; if ((firstIsVowel && secondIsVowel && thirdIsVowel) || (!firstIsVowel && !secondIsVowel && !thirdIsVowel)) { result += (i + 1 - includedLeft) * (str.size() - i - 3 + 1); includedLeft = i + 1; } } return result; } int main() { ios::sync_with_stdio(false); string str; str.reserve(maxN); cin >> str; if (str.size() < 3) { cout << 0; } else { cout << findAllTriples(str); } cout << endl; } |
English