#include<iostream>
#include<vector>
#include<string>
using namespace std;
int main()
{
string speach;
cin>>speach;
int n = (int)speach.size();
vector<int> word(n);
vector<int> tab(n);
long long result = 0;
for(int i = 0; i < n; i++)
{
if(speach[i] == 'a' || speach[i] == 'e' || speach[i] == 'i' || speach[i] == 'o' || speach[i] == 'u' || speach[i] == 'y')
word[i] = -1;
else
word[i] = 1;
}
for(int p = 0; p < n; p++)
{
//for(int i =0; i<n;i++)tab[i]=0;
//if(p != 0) tab[p-1] = 0;
for(int k = p; k < n; k++)
{
if(p == k)
tab[k] = word[p];
else if(tab[k-1] >= 3 || tab[k-1] <= -3)
{
result += (n - k);
break;
}
else
{
if(tab[k-1] > 0 && word[k] == 1)
{
tab[k] = 0;
tab[k] += tab[k-1] + 1;
}
else if(tab[k-1] > 0 && word[k] == -1)
tab[k] = -1;
else if(tab[k-1] < 0 && word[k] == 1)
tab[k] = 1;
else if(tab[k-1] < 0 && word[k] == -1)
{
tab[k] = 0;
tab[k] += tab[k-1] - 1;
}
// if(p == 1 && k == 3) cout<<tab[k-1]<<endl;
if(tab[k] >= 3 || tab[k] <= -3)
{
result++;
}
}
}
//for(int i = 0; i < tab.size(); i++)cout<<tab[i]<<" "; cout<<endl;
}
cout<<result<<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 53 54 55 56 57 58 59 60 | #include<iostream> #include<vector> #include<string> using namespace std; int main() { string speach; cin>>speach; int n = (int)speach.size(); vector<int> word(n); vector<int> tab(n); long long result = 0; for(int i = 0; i < n; i++) { if(speach[i] == 'a' || speach[i] == 'e' || speach[i] == 'i' || speach[i] == 'o' || speach[i] == 'u' || speach[i] == 'y') word[i] = -1; else word[i] = 1; } for(int p = 0; p < n; p++) { //for(int i =0; i<n;i++)tab[i]=0; //if(p != 0) tab[p-1] = 0; for(int k = p; k < n; k++) { if(p == k) tab[k] = word[p]; else if(tab[k-1] >= 3 || tab[k-1] <= -3) { result += (n - k); break; } else { if(tab[k-1] > 0 && word[k] == 1) { tab[k] = 0; tab[k] += tab[k-1] + 1; } else if(tab[k-1] > 0 && word[k] == -1) tab[k] = -1; else if(tab[k-1] < 0 && word[k] == 1) tab[k] = 1; else if(tab[k-1] < 0 && word[k] == -1) { tab[k] = 0; tab[k] += tab[k-1] - 1; } // if(p == 1 && k == 3) cout<<tab[k-1]<<endl; if(tab[k] >= 3 || tab[k] <= -3) { result++; } } } //for(int i = 0; i < tab.size(); i++)cout<<tab[i]<<" "; cout<<endl; } cout<<result<<endl; } |
English