#include <bits/stdc++.h> #define ll long long int #define pb push_back #define mp make_pair #define vi vector<int> #define pii pair<int,int> #define piii pair<int,pii> #define vii vector<pair<int,int> > #define st first #define nd second const int mod=1000000007; const int inf=1000000009; const long long int INF=1000000000000000009; using namespace std; char S[200005]; int funkcja(char c) { if(c=='a'||c=='e'||c=='o'||c=='u'||c=='y'||c=='i') return 1; return 2; } int main() { int n=1; scanf("%s",S+1); while('a'<=S[n]&&S[n]<='z') n++; n--; ll wynik=(ll)n*(n+1)/2; wynik-=n; wynik-=n-1; int last=-1; int ostatni=0; int ile=0; for(int i=1;i<=n;i++) { int x=funkcja(S[i]); if(x==last) ile++; else { if(ile>=3) { ostatni=i-3; } ile=1; } if(ile==3) { int y=i-1-ostatni; wynik-=(ll)y*(y+1)/2; wynik+=y; wynik+=y-1; } last=x; } if(ile<3) { int x=n-ostatni; wynik-=(ll)x*(x+1)/2; wynik+=x; wynik+=x-1; } printf("%lld",wynik); 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 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 | #include <bits/stdc++.h> #define ll long long int #define pb push_back #define mp make_pair #define vi vector<int> #define pii pair<int,int> #define piii pair<int,pii> #define vii vector<pair<int,int> > #define st first #define nd second const int mod=1000000007; const int inf=1000000009; const long long int INF=1000000000000000009; using namespace std; char S[200005]; int funkcja(char c) { if(c=='a'||c=='e'||c=='o'||c=='u'||c=='y'||c=='i') return 1; return 2; } int main() { int n=1; scanf("%s",S+1); while('a'<=S[n]&&S[n]<='z') n++; n--; ll wynik=(ll)n*(n+1)/2; wynik-=n; wynik-=n-1; int last=-1; int ostatni=0; int ile=0; for(int i=1;i<=n;i++) { int x=funkcja(S[i]); if(x==last) ile++; else { if(ile>=3) { ostatni=i-3; } ile=1; } if(ile==3) { int y=i-1-ostatni; wynik-=(ll)y*(y+1)/2; wynik+=y; wynik+=y-1; } last=x; } if(ile<3) { int x=n-ostatni; wynik-=(ll)x*(x+1)/2; wynik+=x; wynik+=x-1; } printf("%lld",wynik); return 0; } |