Robotti

#include <iostream>
#include <unordered_set>
#include <cmath>
#include <vector>
 
int main() {
    int n;
    std::string rooms;
    std::cin >> n >> rooms;
    //Lisää laadukkaita määritelmiä
    int start_position = -1;
    std::unordered_set<int> coin_positions;
    //Jos ei toimi, siirry eteenpäin
    for (int i = 0; i < n; ++i) {
        if (rooms[i] == 'R') {
            start_position = i;
        } else if (rooms[i] == '*') {
            coin_positions.insert(i);
        }
    }
    
    int steps = 0;
    int coins_collected = 0;
    int current_position = start_position;
    //Mukavat while ja sen alku
    while (!coin_positions.empty()) {
        int min_distance = n;
        int closest_coin_position = -1;
        bool found_duplicate = false;
        
        for (int pos : coin_positions) {
            int distance = std::abs(current_position - pos);
            if (distance < min_distance) {
                min_distance = distance;
                closest_coin_position = pos;
                found_duplicate = false;
            } else if (distance == min_distance) {
                found_duplicate = true;
            }
        }
        
        if (found_duplicate) {
            break;
        }
 
        steps += min_distance;
        coins_collected += 1;
        current_position = closest_coin_position;
        
        coin_positions.erase(closest_coin_position);
        //Whilen loppu
    }
    
    std::cout << steps << " " << coins_collected << 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