#include<bits/stdc++.h> #define rep(i,k,n) for(ll i= (ll) k;i< (ll) n;i++) #define all(v) (v).begin(), (v).end() #define SZ(v) (ll)((v).size()) #define pb push_back #define ft first #define sd second typedef long long ll; typedef unsigned long long ull; typedef long double ld; const long long INF = 1e18L + 1; const int IINF = 1e9 + 1; using namespace std; template<class TH> void _dbg(const char *sdbg, TH h){ cerr<<sdbg<<'='<<h<<endl; } template<class TH, class... TA> void _dbg(const char *sdbg, TH h, TA... a) { while(*sdbg!=',')cerr<<*sdbg++; cerr<<'='<<h<<','; _dbg(sdbg+1, a...); } #ifdef LOCAL #define DBG(...) _dbg(#__VA_ARGS__, __VA_ARGS__) #else #define DBG(...) (__VA_ARGS__) #define cerr if(0)cout #endif const vector<char>vow = {'a', 'e', 'i', 'o', 'u', 'y'}; int main() { #ifndef LOCAL ios_base::sync_with_stdio(0); cin.tie(0); #endif string s; cin >> s; vector<ll>vow_count(SZ(s)), bad; transform(all(s), vow_count.begin(), [&](char c){return count(all(vow), c);}); rep(i, 0, SZ(s)){ if(i == 0 or i == SZ(s) - 1){ bad.pb(i); } else{ ll local_vow_cnt = vow_count[i - 1] + vow_count[i] + vow_count[i + 1]; if(local_vow_cnt == 0 or local_vow_cnt == 3){ bad.pb(i); } } } ll res = SZ(s) * (SZ(s) - 1) / 2; rep(i, 1, SZ(bad)){ ll len = bad[i] - bad[i - 1] + 1; res -= len * (len - 1) / 2; } cout << res << "\n"; 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 | #include<bits/stdc++.h> #define rep(i,k,n) for(ll i= (ll) k;i< (ll) n;i++) #define all(v) (v).begin(), (v).end() #define SZ(v) (ll)((v).size()) #define pb push_back #define ft first #define sd second typedef long long ll; typedef unsigned long long ull; typedef long double ld; const long long INF = 1e18L + 1; const int IINF = 1e9 + 1; using namespace std; template<class TH> void _dbg(const char *sdbg, TH h){ cerr<<sdbg<<'='<<h<<endl; } template<class TH, class... TA> void _dbg(const char *sdbg, TH h, TA... a) { while(*sdbg!=',')cerr<<*sdbg++; cerr<<'='<<h<<','; _dbg(sdbg+1, a...); } #ifdef LOCAL #define DBG(...) _dbg(#__VA_ARGS__, __VA_ARGS__) #else #define DBG(...) (__VA_ARGS__) #define cerr if(0)cout #endif const vector<char>vow = {'a', 'e', 'i', 'o', 'u', 'y'}; int main() { #ifndef LOCAL ios_base::sync_with_stdio(0); cin.tie(0); #endif string s; cin >> s; vector<ll>vow_count(SZ(s)), bad; transform(all(s), vow_count.begin(), [&](char c){return count(all(vow), c);}); rep(i, 0, SZ(s)){ if(i == 0 or i == SZ(s) - 1){ bad.pb(i); } else{ ll local_vow_cnt = vow_count[i - 1] + vow_count[i] + vow_count[i + 1]; if(local_vow_cnt == 0 or local_vow_cnt == 3){ bad.pb(i); } } } ll res = SZ(s) * (SZ(s) - 1) / 2; rep(i, 1, SZ(bad)){ ll len = bad[i] - bad[i - 1] + 1; res -= len * (len - 1) / 2; } cout << res << "\n"; return 0; } |