// 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; } |