#include <bits/stdc++.h> // #define MULTIPLE_TESTS // #define ENDLESS_TESTS #define TIME_LIMIT 2 #define MEMORY_LIMIT 512 using namespace std; void test() { std::string text; text.reserve(200'001); std::cin >> text; std::vector<int> bs; bs.reserve(200'001); const int length = text.size(); for (int i = 0; i < length; ++i) if (text[i] == 'b') bs.push_back(i); if (length % 2 == 0 && bs.size()%2 == 1) { std::cout << "-1\n"; return; } long long answer = 0; const int pairs = bs.size() / 2; auto it_front = bs.begin(); auto it_back = bs.rbegin(); for (int i = 0; i < pairs; ++i) { const auto lhs = *(it_front++); const auto rhs = length - *(it_back++) - 1; answer += std::abs(lhs - rhs); } if (bs.size() % 2 == 1) { const auto val = *it_front; const auto mid = length / 2; answer += std::abs(val - mid); } std::cout << answer << '\n'; } int main() { #ifndef CONTEST_WORKSPACE std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); #endif #ifdef ENDLESS_TESTS while(!(cin >> std::ws).eof()) test(); #else int T = 0; #ifdef MULTIPLE_TESTS cin >> T; #else T = 1; #endif while (T --> 0) test(); #endif 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 | #include <bits/stdc++.h> // #define MULTIPLE_TESTS // #define ENDLESS_TESTS #define TIME_LIMIT 2 #define MEMORY_LIMIT 512 using namespace std; void test() { std::string text; text.reserve(200'001); std::cin >> text; std::vector<int> bs; bs.reserve(200'001); const int length = text.size(); for (int i = 0; i < length; ++i) if (text[i] == 'b') bs.push_back(i); if (length % 2 == 0 && bs.size()%2 == 1) { std::cout << "-1\n"; return; } long long answer = 0; const int pairs = bs.size() / 2; auto it_front = bs.begin(); auto it_back = bs.rbegin(); for (int i = 0; i < pairs; ++i) { const auto lhs = *(it_front++); const auto rhs = length - *(it_back++) - 1; answer += std::abs(lhs - rhs); } if (bs.size() % 2 == 1) { const auto val = *it_front; const auto mid = length / 2; answer += std::abs(val - mid); } std::cout << answer << '\n'; } int main() { #ifndef CONTEST_WORKSPACE std::ios_base::sync_with_stdio(false); std::cin.tie(nullptr); #endif #ifdef ENDLESS_TESTS while(!(cin >> std::ws).eof()) test(); #else int T = 0; #ifdef MULTIPLE_TESTS cin >> T; #else T = 1; #endif while (T --> 0) test(); #endif return 0; } |