Robotti

input/code.cpp: In function 'int main()':
input/code.cpp:15:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |     for (int i = 0; i < layout.length(); ++i) {
      |                     ~~^~~~~~~~~~~~~~~~~

#include <iostream>
#include <string>
#include <set>
#include <algorithm>
#include <cmath>

int main() {
    int Num_rooms;
    std::string layout;
    std::cin >> Num_rooms;
    std::cin >> layout;

    int robot_pos = layout.find('R');
    std::set<int> star_positions;
    for (int i = 0; i < layout.length(); ++i) {
        if (layout[i] == '*') {
            star_positions.insert(i);
        }
    }

    int total_steps = 0;
    int total_coins = 0;

    while (!star_positions.empty()) {
        auto closest_star = std::min_element(star_positions.begin(), star_positions.end(),
            [robot_pos](int a, int b) {
                return std::abs(a - robot_pos) < std::abs(b - robot_pos);
            });
        
        int distance = std::abs(*closest_star - robot_pos);

        bool multiple_stars_at_same_distance = false;
        for (int star : star_positions) {
            if (star != *closest_star && std::abs(star - robot_pos) == distance) {
                multiple_stars_at_same_distance = true;
                break;
            }
        }

        if (multiple_stars_at_same_distance) {
            break;
        }

        total_steps += distance;
        total_coins++;
        robot_pos = *closest_star;
        star_positions.erase(closest_star);
    }

    std::cout << total_steps << " " << total_coins << std::endl;

    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: ACCEPTED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 1, 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 1, 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 1, 2

Verdict: ACCEPTED

Test 12

Group: 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED

Test 14

Group: 2

Verdict: ACCEPTED

Test 15

Group: 2

Verdict: ACCEPTED

Test 16

Group: 2

Verdict: ACCEPTED

Test 17

Group: 2

Verdict: ACCEPTED

Test 18

Group: 2

Verdict: ACCEPTED

Test 19

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: ACCEPTED

Test 23

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 2

Verdict: TIME LIMIT EXCEEDED