// Jakub Rozek
// Palindrom
// czas: n*log(n)
// pami: n
#include "bits/stdc++.h"
using namespace std;
const int R=131072;
string s;
int n,a,b,aa,bb,nr,srodek,dwaczesc;
long long odp;
int tree[R*2];
stack <int> sa, sb;
vector <int> v;
void dodaj(int a)
{
a+=R;
while(a)
{
++tree[a];
a/=2;
}
}
int ile(int w, int a, int b, int c, int d)
{
if(a>d || b<c) return 0;
if(a>=c && b<=d) return tree[w];
return ile(w*2,a,(a+b)/2,c,d)+ile(w*2+1,(a+b)/2+1,b,c,d);
}
void inwersje()
{
for(int i:v)
{
odp+=ile(1,0,R-1,i,R-1);
dodaj(i);
}
}
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cin>>s;
n=s.size();
for(int i=0; i<n; ++i)
{
if(s[i]=='a') ++a;
else ++b;
}
if(n%2==0 && a%2==1)
{
cout<<"-1\n";
return 0;
}
//dzieli na pocz i koniec z numeracja na jakiej pozycji powinny byc
for(int i=0; i<n; ++i)
{
if(s[i]=='a')
{
++aa;
if(aa*2-1==a)
{
srodek=1;
odp+=dwaczesc;
continue;
}
if(aa*2<=a)
{
++nr;
sa.push(nr);
odp+=dwaczesc+srodek;
}
else
{
++dwaczesc;
v.push_back(sa.top());
sa.pop();
}
}
else
{
++bb;
if(bb*2-1==b)
{
srodek=1;
odp+=dwaczesc;
continue;
}
if(bb*2<=b)
{
++nr;
sb.push(nr);
odp+=dwaczesc+srodek;
}
else
{
++dwaczesc;
v.push_back(sb.top());
sb.pop();
}
}
}
reverse(v.begin(), v.end());
inwersje();
cout<<odp<<"\n";
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 | // Jakub Rozek // Palindrom // czas: n*log(n) // pami: n #include "bits/stdc++.h" using namespace std; const int R=131072; string s; int n,a,b,aa,bb,nr,srodek,dwaczesc; long long odp; int tree[R*2]; stack <int> sa, sb; vector <int> v; void dodaj(int a) { a+=R; while(a) { ++tree[a]; a/=2; } } int ile(int w, int a, int b, int c, int d) { if(a>d || b<c) return 0; if(a>=c && b<=d) return tree[w]; return ile(w*2,a,(a+b)/2,c,d)+ile(w*2+1,(a+b)/2+1,b,c,d); } void inwersje() { for(int i:v) { odp+=ile(1,0,R-1,i,R-1); dodaj(i); } } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin>>s; n=s.size(); for(int i=0; i<n; ++i) { if(s[i]=='a') ++a; else ++b; } if(n%2==0 && a%2==1) { cout<<"-1\n"; return 0; } //dzieli na pocz i koniec z numeracja na jakiej pozycji powinny byc for(int i=0; i<n; ++i) { if(s[i]=='a') { ++aa; if(aa*2-1==a) { srodek=1; odp+=dwaczesc; continue; } if(aa*2<=a) { ++nr; sa.push(nr); odp+=dwaczesc+srodek; } else { ++dwaczesc; v.push_back(sa.top()); sa.pop(); } } else { ++bb; if(bb*2-1==b) { srodek=1; odp+=dwaczesc; continue; } if(bb*2<=b) { ++nr; sb.push(nr); odp+=dwaczesc+srodek; } else { ++dwaczesc; v.push_back(sb.top()); sb.pop(); } } } reverse(v.begin(), v.end()); inwersje(); cout<<odp<<"\n"; return 0; } |
English