#include<bits/stdc++.h> using namespace std; const int N = 1048576 + 1; vector<int> L; priority_queue<int> P; int t[N]; int n, base = 1; void dodaj(int a, int b, int wartosc) { a += base; b += base; t[a] += wartosc; if(a != b) t[b] += wartosc; while(a/2 != b/2) { if(a % 2 == 0) t[a + 1] += wartosc; if(b % 2 == 1) t[b - 1] += wartosc; a /= 2; b /= 2; } } int zapytanie(int a) { int wynik = 0; a += base; while(a > 0) { wynik += t[a]; a /= 2; } return wynik; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin >> n; while(base < n) base *= 2; bool bylo = false; for(int i = 0; i < n; i ++) { char y; cin >> y; if(y == 'L' && bylo) { L.push_back(i); } else { if(y == 'P') { bylo = true; P.push(-i); } } } if(L.empty()) { for(int i = 0; i < n; i ++) { cout << 0 << " "; } return 0; } for(auto l : L) { int p = - P.top(); if(p > l) continue; dodaj(p, l, 1); if(p + 1 <= l - 1) { dodaj(p + 1, l - 1, 1); } P.pop(); P.push(-l); } for(int i = 0; i < n; i ++) { cout << zapytanie(i) << " "; } cout << "\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 | #include<bits/stdc++.h> using namespace std; const int N = 1048576 + 1; vector<int> L; priority_queue<int> P; int t[N]; int n, base = 1; void dodaj(int a, int b, int wartosc) { a += base; b += base; t[a] += wartosc; if(a != b) t[b] += wartosc; while(a/2 != b/2) { if(a % 2 == 0) t[a + 1] += wartosc; if(b % 2 == 1) t[b - 1] += wartosc; a /= 2; b /= 2; } } int zapytanie(int a) { int wynik = 0; a += base; while(a > 0) { wynik += t[a]; a /= 2; } return wynik; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin >> n; while(base < n) base *= 2; bool bylo = false; for(int i = 0; i < n; i ++) { char y; cin >> y; if(y == 'L' && bylo) { L.push_back(i); } else { if(y == 'P') { bylo = true; P.push(-i); } } } if(L.empty()) { for(int i = 0; i < n; i ++) { cout << 0 << " "; } return 0; } for(auto l : L) { int p = - P.top(); if(p > l) continue; dodaj(p, l, 1); if(p + 1 <= l - 1) { dodaj(p + 1, l - 1, 1); } P.pop(); P.push(-l); } for(int i = 0; i < n; i ++) { cout << zapytanie(i) << " "; } cout << "\n"; return 0; } |