#include <bits/stdc++.h>
using namespace std;
int main(){
ios_base::sync_with_stdio(0);
cout.tie(0); cin.tie(0);
int n; cin>>n;
int zerocount = 0;
vector<bool> in(n, 0); for(int i=0; i<n; i++){char x; cin>>x; if(x == 'P') in[i] = 1; else zerocount++;}
//-----------------------------
if(zerocount == 0 || zerocount == n){
for(int i=0; i<n; i++) cout<<"0 ";
return 0;
}
//-----------------------------
vector<int> zeropref(n, 0); zeropref[0] = !in[0]; for(int i=1; i<n; i++) zeropref[i] = zeropref[i-1] + !in[i];
//-----------------------------
vector<int> results(n, 0);
for(int i=0; i<zerocount; i++){
results[i] = 2 * (i + 1 - zeropref[i]); if(in[i] == 1) results[i]--;
}
for(int i=zerocount; i<n; i++){
results[i] = 2 * (zeropref[n-1] - zeropref[i-1]); if(in[i] == 0) results[i]--;
}
//-----------------------------
for(auto e: results) cout<<e<<" ";
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 | #include <bits/stdc++.h> using namespace std; int main(){ ios_base::sync_with_stdio(0); cout.tie(0); cin.tie(0); int n; cin>>n; int zerocount = 0; vector<bool> in(n, 0); for(int i=0; i<n; i++){char x; cin>>x; if(x == 'P') in[i] = 1; else zerocount++;} //----------------------------- if(zerocount == 0 || zerocount == n){ for(int i=0; i<n; i++) cout<<"0 "; return 0; } //----------------------------- vector<int> zeropref(n, 0); zeropref[0] = !in[0]; for(int i=1; i<n; i++) zeropref[i] = zeropref[i-1] + !in[i]; //----------------------------- vector<int> results(n, 0); for(int i=0; i<zerocount; i++){ results[i] = 2 * (i + 1 - zeropref[i]); if(in[i] == 1) results[i]--; } for(int i=zerocount; i<n; i++){ results[i] = 2 * (zeropref[n-1] - zeropref[i-1]); if(in[i] == 0) results[i]--; } //----------------------------- for(auto e: results) cout<<e<<" "; return 0; } |
English