#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; } |
English