#include <cstdio> #include <cstring> char samogloski[] = {'a', 'e', 'i', 'o', 'u', 'y'}; char text[200001]; bool isSam(char c) { for(int i = 0; i < 6; ++i) { if(c == samogloski[i]) { return true; } } return false; } unsigned long long calc(int pre, int post, int len) { // printf("calc: %d, %d, %d\n", pre, post, len); unsigned long long res = 0; res += pre*post; if(len > 3) { res += pre * (len-3); res += post * (len-3); if(len >= 5) { for(int i = 3; i <= len - 2; ++i) { res += 1 + (len-2) - i; } } } return (unsigned long long)res; } int findNext(int& pos, int size) { int sam = 0; int spu = 0; bool wasSam = false; // printf("findNext: %d\n", pos); for(; pos < size; ++pos) { if(isSam(text[pos])) { ++sam; if(!wasSam) { if(spu > 2) { return spu; } spu = 0; } wasSam = true; } else { ++spu; if(wasSam) { if(sam > 2) { return sam; } sam = 0; } wasSam = false; } } if(wasSam) { if(sam > 2) { return sam; } } else { if(spu > 2) { return spu; } } return -1; } int main(void) { int size = 0; char buff; int sam = 0; int spu = 0; bool wasSam = false; unsigned long long result = 0; scanf("%s", text); size = strlen(text); int pos = 0; int len = findNext(pos, size); if(len == -1) { printf("0"); return 0; } while(true) { int currentPos = pos; int nextLen = findNext(pos, size); // printf("nextLen: %d pos: %d\n", nextLen, pos); if(pos == currentPos || nextLen == -1) ++pos; result += calc(currentPos - len + 1, pos - currentPos, len); if(nextLen != -1) { len = nextLen; } else { break; } } // int i; // for(i = word.second ; i < size; ++i) { // if(isSam(text[i])) { // ++sam; // if(!wasSam) { // if(spu > 2) { // result += calc(i-spu, size-i, spu); // } // spu = 0; // } // wasSam = true; // } else { // ++spu; // if(wasSam) { // if(sam > 2) { // result += calc(i-sam, size-i, sam); // } // sam = 0; // } // wasSam = false; // } // } // if(wasSam) { // if(sam > 2) { // result += calc(i-sam, size-i, sam); // } // } else { // if(spu > 2) { // result += calc(i-spu, size-i, spu); // } // } printf("%ld\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 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 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 | #include <cstdio> #include <cstring> char samogloski[] = {'a', 'e', 'i', 'o', 'u', 'y'}; char text[200001]; bool isSam(char c) { for(int i = 0; i < 6; ++i) { if(c == samogloski[i]) { return true; } } return false; } unsigned long long calc(int pre, int post, int len) { // printf("calc: %d, %d, %d\n", pre, post, len); unsigned long long res = 0; res += pre*post; if(len > 3) { res += pre * (len-3); res += post * (len-3); if(len >= 5) { for(int i = 3; i <= len - 2; ++i) { res += 1 + (len-2) - i; } } } return (unsigned long long)res; } int findNext(int& pos, int size) { int sam = 0; int spu = 0; bool wasSam = false; // printf("findNext: %d\n", pos); for(; pos < size; ++pos) { if(isSam(text[pos])) { ++sam; if(!wasSam) { if(spu > 2) { return spu; } spu = 0; } wasSam = true; } else { ++spu; if(wasSam) { if(sam > 2) { return sam; } sam = 0; } wasSam = false; } } if(wasSam) { if(sam > 2) { return sam; } } else { if(spu > 2) { return spu; } } return -1; } int main(void) { int size = 0; char buff; int sam = 0; int spu = 0; bool wasSam = false; unsigned long long result = 0; scanf("%s", text); size = strlen(text); int pos = 0; int len = findNext(pos, size); if(len == -1) { printf("0"); return 0; } while(true) { int currentPos = pos; int nextLen = findNext(pos, size); // printf("nextLen: %d pos: %d\n", nextLen, pos); if(pos == currentPos || nextLen == -1) ++pos; result += calc(currentPos - len + 1, pos - currentPos, len); if(nextLen != -1) { len = nextLen; } else { break; } } // int i; // for(i = word.second ; i < size; ++i) { // if(isSam(text[i])) { // ++sam; // if(!wasSam) { // if(spu > 2) { // result += calc(i-spu, size-i, spu); // } // spu = 0; // } // wasSam = true; // } else { // ++spu; // if(wasSam) { // if(sam > 2) { // result += calc(i-sam, size-i, sam); // } // sam = 0; // } // wasSam = false; // } // } // if(wasSam) { // if(sam > 2) { // result += calc(i-sam, size-i, sam); // } // } else { // if(spu > 2) { // result += calc(i-spu, size-i, spu); // } // } printf("%ld\n", result); return 0; } |