#include <bits/stdc++.h> using namespace std; typedef long long int LL; void normalise(vector<int>& vec) { const int minElem = *min_element(vec.cbegin(), vec.cend()); for(int i = 0; i < vec.size(); ++i) { vec[i] -= minElem; } } LL countSubstrings(const string& text, unordered_set<char> allowedChars) { unordered_map<char, int> charToPos; int pos = 0; for(const char c : allowedChars) { charToPos[c] = pos++; } map<vector<int>, LL> statesCount; vector<int> charsCount(charToPos.size(), 0); ++statesCount[charsCount]; for(const char c : text) { if(allowedChars.find(c) == allowedChars.cend()) { continue; } ++charsCount[charToPos[c]]; normalise(charsCount); ++statesCount[charsCount]; } LL res = 0; for(const auto& elem : statesCount) { res += (elem.second * (elem.second-1) / 2); } return res; } vector<string> split(const string& text, unordered_set<char> allowedChars) { vector<string> res; string curr; unordered_set<char> missingChars = allowedChars; for(char c : text) { if(allowedChars.find(c) == allowedChars.cend()) { if(missingChars.empty() && !curr.empty()) { res.push_back(curr); } missingChars = allowedChars; curr = ""; continue; } curr += c; missingChars.erase(c); } if(missingChars.empty() && !curr.empty()) { res.push_back(curr); } return res; } LL countSubstrings(const string& s) { const vector<unordered_set<char>> allowedCharsSet = { {'a'}, {'b'}, {'c'}, {'a', 'b'}, {'a', 'c'}, {'b', 'c'}, {'a', 'b', 'c'} }; LL res = 0; for(const auto& allowedChars : allowedCharsSet) { vector<string> chunks = split(s, allowedChars); for(const string& chunk : chunks) { res += countSubstrings(chunk, allowedChars); } } return res; } int main() { string s; cin >> s; cout << countSubstrings(s) << endl; 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 | #include <bits/stdc++.h> using namespace std; typedef long long int LL; void normalise(vector<int>& vec) { const int minElem = *min_element(vec.cbegin(), vec.cend()); for(int i = 0; i < vec.size(); ++i) { vec[i] -= minElem; } } LL countSubstrings(const string& text, unordered_set<char> allowedChars) { unordered_map<char, int> charToPos; int pos = 0; for(const char c : allowedChars) { charToPos[c] = pos++; } map<vector<int>, LL> statesCount; vector<int> charsCount(charToPos.size(), 0); ++statesCount[charsCount]; for(const char c : text) { if(allowedChars.find(c) == allowedChars.cend()) { continue; } ++charsCount[charToPos[c]]; normalise(charsCount); ++statesCount[charsCount]; } LL res = 0; for(const auto& elem : statesCount) { res += (elem.second * (elem.second-1) / 2); } return res; } vector<string> split(const string& text, unordered_set<char> allowedChars) { vector<string> res; string curr; unordered_set<char> missingChars = allowedChars; for(char c : text) { if(allowedChars.find(c) == allowedChars.cend()) { if(missingChars.empty() && !curr.empty()) { res.push_back(curr); } missingChars = allowedChars; curr = ""; continue; } curr += c; missingChars.erase(c); } if(missingChars.empty() && !curr.empty()) { res.push_back(curr); } return res; } LL countSubstrings(const string& s) { const vector<unordered_set<char>> allowedCharsSet = { {'a'}, {'b'}, {'c'}, {'a', 'b'}, {'a', 'c'}, {'b', 'c'}, {'a', 'b', 'c'} }; LL res = 0; for(const auto& allowedChars : allowedCharsSet) { vector<string> chunks = split(s, allowedChars); for(const string& chunk : chunks) { res += countSubstrings(chunk, allowedChars); } } return res; } int main() { string s; cin >> s; cout << countSubstrings(s) << endl; return 0; } |