#include <cstdio>
// Marcin Regdos | Potyczki Algorytmiczne 2018 | Runda 1B POL Język polski
int main(){
char input[200005];
bool inputb[200005];
long long result = 0;
int n = 0, previd = -1;
scanf ("%s", input);
bool alphabet ['z' + 1];
for (int i = 'a'; i <= static_cast<int>('z'); ++i) alphabet [i] = false;
alphabet['a'] = alphabet['e'] = alphabet['y'] = alphabet['i'] = alphabet['u'] = alphabet['o'] = true;
for ( n = 0; input[n]; ++n)
{
inputb[n] = alphabet[input[n]];
}
for (int i = 0; i + 2 < n; ++i)
{
if ( (inputb[i] && inputb[i+1] && inputb[i+2])
|| (!inputb[i] && !inputb[i+1] && !inputb[i+2]) )
{
result += static_cast<long long>( i - previd ) * static_cast<long long>( n - (i + 2)) ;
previd = i;
}
}
printf ("%lld\n", result);
return 0;
}
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 | #include <cstdio> // Marcin Regdos | Potyczki Algorytmiczne 2018 | Runda 1B POL Język polski int main(){ char input[200005]; bool inputb[200005]; long long result = 0; int n = 0, previd = -1; scanf ("%s", input); bool alphabet ['z' + 1]; for (int i = 'a'; i <= static_cast<int>('z'); ++i) alphabet [i] = false; alphabet['a'] = alphabet['e'] = alphabet['y'] = alphabet['i'] = alphabet['u'] = alphabet['o'] = true; for ( n = 0; input[n]; ++n) { inputb[n] = alphabet[input[n]]; } for (int i = 0; i + 2 < n; ++i) { if ( (inputb[i] && inputb[i+1] && inputb[i+2]) || (!inputb[i] && !inputb[i+1] && !inputb[i+2]) ) { result += static_cast<long long>( i - previd ) * static_cast<long long>( n - (i + 2)) ; previd = i; } } printf ("%lld\n", result); return 0; } |
English