#include <bits/stdc++.h>
using namespace std;
#define FOR(i,a,b) for(int i=a; i<=b; ++i)
constexpr int baseN =1<<19;
int N, rres[2][2*baseN], posEnd[baseN];
bitset<baseN> inp;
int GetSum(int v, bool dim){
v+=baseN;
int res=0;
while(v){
res+=rres[dim][v];
v/=2;
}
return res;
}
void Add(int p, int q, int val, int dim){
p+=baseN-1, q+=baseN+1;
while(p/2!=q/2){
if(p%2==0) rres[dim][p+1]+=val;
if(q%2==1) rres[dim][q-1]+=val;
q/=2, p/=2;
}
}
void Input(){
string cInp;
int cPos=0;
cin>>N>>cInp;
FOR(i,0,N-1){
if(cInp[i]=='P') inp[i]=1;
else posEnd[i]=cPos++;
}
FOR(i,0,N-1)
if(inp[i]) posEnd[i]=cPos++;
}
void Print(){
FOR(i,0,N-1)
cout<<GetSum(i,inp[i])<<' ';
cout<<'\n';
}
void Solve(){
FOR(i,0,N-1){
if(inp[i]){
Add(i+1,posEnd[i]-1,2,0);
Add(posEnd[i],posEnd[i],1,0);
}else{
Add(posEnd[i]+1,i-1,2,1);
Add(posEnd[i],posEnd[i],1,1);
}
}
}
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
Input();
Solve();
Print();
}
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 | #include <bits/stdc++.h> using namespace std; #define FOR(i,a,b) for(int i=a; i<=b; ++i) constexpr int baseN =1<<19; int N, rres[2][2*baseN], posEnd[baseN]; bitset<baseN> inp; int GetSum(int v, bool dim){ v+=baseN; int res=0; while(v){ res+=rres[dim][v]; v/=2; } return res; } void Add(int p, int q, int val, int dim){ p+=baseN-1, q+=baseN+1; while(p/2!=q/2){ if(p%2==0) rres[dim][p+1]+=val; if(q%2==1) rres[dim][q-1]+=val; q/=2, p/=2; } } void Input(){ string cInp; int cPos=0; cin>>N>>cInp; FOR(i,0,N-1){ if(cInp[i]=='P') inp[i]=1; else posEnd[i]=cPos++; } FOR(i,0,N-1) if(inp[i]) posEnd[i]=cPos++; } void Print(){ FOR(i,0,N-1) cout<<GetSum(i,inp[i])<<' '; cout<<'\n'; } void Solve(){ FOR(i,0,N-1){ if(inp[i]){ Add(i+1,posEnd[i]-1,2,0); Add(posEnd[i],posEnd[i],1,0); }else{ Add(posEnd[i]+1,i-1,2,1); Add(posEnd[i],posEnd[i],1,1); } } } int main(){ ios_base::sync_with_stdio(0); cin.tie(0), cout.tie(0); Input(); Solve(); Print(); } |
English