// Arti1990, II UWr
#include <bits/stdc++.h>
#define forr(i, n) for(int i=0; i<n; i++)
#define FOREACH(iter, coll) for(auto iter = coll.begin(); iter != coll.end(); ++iter)
#define FOREACHR(iter, coll) for(auto iter = coll.rbegin(); iter != coll.rend(); ++iter)
#define lbound(P,K,FUN) ({auto SSS=P, PPP = P-1, KKK=(K)+1; while(PPP+1!=KKK) {SSS = (PPP+(KKK-PPP)/2); if(FUN(SSS)) KKK = SSS; else PPP = SSS;} PPP;})
#define testy() int _tests; scanf("%d", &_tests); FOR(_test, 1, _tests)
#define CLEAR(tab) memset(tab, 0, sizeof(tab))
#define CONTAIN(el, coll) (coll.find(el) != coll.end())
#define FOR(i, a, b) for(int i=a; i<=b; i++)
#define FORD(i, a, b) for(int i=a; i>=b; i--)
#define MP make_pair
#define PB push_back
#define ff first
#define ss second
#define deb(X) X;
#define M 1000000007
#define INF 1000000007LL
using namespace std;
int n, ilea, ileb;
char slowo[1000007];
long long solve()
{
scanf("%s", slowo);
n = strlen(slowo);
forr(i, n)
{
ilea += slowo[i] == 'a';
ileb += slowo[i] == 'b';
}
char c = (ilea%2 == 0 ? 'a' : (ileb%2 == 0 ? 'b' : -1));
if(c == -1)
return -1;
if(ilea == 0 || ileb == 0)
return 0;
long long res = 0;
int p = 0, k = n-1;
while(true)
{
while(slowo[p] != c)
p++;
while(slowo[k] != c)
k--;
if(k <= p)
return res;
res += abs(n-1-(p+k));
p++;
k--;
}
return 0;
}
int main()
{
printf("%lld\n", solve());
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 | // Arti1990, II UWr #include <bits/stdc++.h> #define forr(i, n) for(int i=0; i<n; i++) #define FOREACH(iter, coll) for(auto iter = coll.begin(); iter != coll.end(); ++iter) #define FOREACHR(iter, coll) for(auto iter = coll.rbegin(); iter != coll.rend(); ++iter) #define lbound(P,K,FUN) ({auto SSS=P, PPP = P-1, KKK=(K)+1; while(PPP+1!=KKK) {SSS = (PPP+(KKK-PPP)/2); if(FUN(SSS)) KKK = SSS; else PPP = SSS;} PPP;}) #define testy() int _tests; scanf("%d", &_tests); FOR(_test, 1, _tests) #define CLEAR(tab) memset(tab, 0, sizeof(tab)) #define CONTAIN(el, coll) (coll.find(el) != coll.end()) #define FOR(i, a, b) for(int i=a; i<=b; i++) #define FORD(i, a, b) for(int i=a; i>=b; i--) #define MP make_pair #define PB push_back #define ff first #define ss second #define deb(X) X; #define M 1000000007 #define INF 1000000007LL using namespace std; int n, ilea, ileb; char slowo[1000007]; long long solve() { scanf("%s", slowo); n = strlen(slowo); forr(i, n) { ilea += slowo[i] == 'a'; ileb += slowo[i] == 'b'; } char c = (ilea%2 == 0 ? 'a' : (ileb%2 == 0 ? 'b' : -1)); if(c == -1) return -1; if(ilea == 0 || ileb == 0) return 0; long long res = 0; int p = 0, k = n-1; while(true) { while(slowo[p] != c) p++; while(slowo[k] != c) k--; if(k <= p) return res; res += abs(n-1-(p+k)); p++; k--; } return 0; } int main() { printf("%lld\n", solve()); return 0; } |
English