#include <iostream>
#include <vector>
int main() {
int64_t n;
std::string desc;
std::cin >> n >> desc;
std::vector<int64_t> left_pressure(n), right_pressure(n);
{
int i = 0;
auto desc_it = desc.begin();
auto left_it = left_pressure.begin();
for (; desc_it != desc.end(); ++desc_it, ++left_it) {
*left_it = i;
if (*desc_it == 'P') {
i += 1;
}
}
}
{
int i = 0;
auto desc_rit = desc.rbegin();
auto right_rit = right_pressure.rbegin();
for (; desc_rit != desc.rend(); ++desc_rit, ++right_rit) {
*right_rit = i;
if (*desc_rit == 'L') {
i += 1;
}
}
}
{
auto desc_it = desc.begin();
auto left_it = left_pressure.begin();
auto right_it = right_pressure.begin();
for (; desc_it != desc.end(); ++desc_it, ++left_it, ++right_it) {
if (*desc_it == 'L') {
if (*right_it > *left_it) {
*right_it = *left_it;
} else if (*left_it > *right_it + 1) {
*left_it = *right_it + 1;
}
} else {
if (*left_it > *right_it) {
*left_it = *right_it;
} else if (*right_it > *left_it + 1) {
*right_it = *left_it + 1;
}
}
std::cout << *left_it + *right_it << " ";
}
std::cout << "\n";
}
}
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 | #include <iostream> #include <vector> int main() { int64_t n; std::string desc; std::cin >> n >> desc; std::vector<int64_t> left_pressure(n), right_pressure(n); { int i = 0; auto desc_it = desc.begin(); auto left_it = left_pressure.begin(); for (; desc_it != desc.end(); ++desc_it, ++left_it) { *left_it = i; if (*desc_it == 'P') { i += 1; } } } { int i = 0; auto desc_rit = desc.rbegin(); auto right_rit = right_pressure.rbegin(); for (; desc_rit != desc.rend(); ++desc_rit, ++right_rit) { *right_rit = i; if (*desc_rit == 'L') { i += 1; } } } { auto desc_it = desc.begin(); auto left_it = left_pressure.begin(); auto right_it = right_pressure.begin(); for (; desc_it != desc.end(); ++desc_it, ++left_it, ++right_it) { if (*desc_it == 'L') { if (*right_it > *left_it) { *right_it = *left_it; } else if (*left_it > *right_it + 1) { *left_it = *right_it + 1; } } else { if (*left_it > *right_it) { *left_it = *right_it; } else if (*right_it > *left_it + 1) { *right_it = *left_it + 1; } } std::cout << *left_it + *right_it << " "; } std::cout << "\n"; } } |
English