Robotti

#include <iostream>
#include <string>
#include <set>
#include <algorithm>
#include <bits/stdc++.h> 

unsigned int robottiIndeksi;

int etsi(std::set<int> kolikotX, int robottiX){

    if (kolikotX.size() <= 1){
        return 0;
    }
    auto it = kolikotX.begin();
    if (robottiIndeksi == 0){
        std::advance(it, 1);
        return *it - robottiX;
    } else if (robottiIndeksi == (kolikotX.size() - 1)){
        robottiIndeksi--;
        std::advance(it, kolikotX.size() - 2);
        return *it - robottiX;
    }
    std::advance(it, robottiIndeksi - 1);
    int vas = abs(*it - robottiX);
    std::advance(it, 2);
    int oik = abs(*it - robottiX);
    

    if (oik < vas) {
        return oik;
    } else if (vas < oik) {
        robottiIndeksi--;
        return -vas;
    } else {
        return 0;
    }
}

int main() { 
    int n;
    std::cin >> n;
    std::string map;
    std::getline(std::cin, map);
    std::cin >> map;
    
    std::set<int> kolikotX;
    int robottiX = 0;
    for (unsigned int i = 0; i < map.size(); i++) {
        if (map[i] == '*'){
            kolikotX.insert(i);
            continue;
        } else if (map[i] == 'R'){
            robottiX = i;
            kolikotX.insert(i);
            robottiIndeksi = kolikotX.size() - 1;
        }
    }

    int askelia = 0;
    int kolikoita = 0;
    while (true){
        int et = etsi(kolikotX, robottiX);
        if (et == 0){
            break;
        }
        askelia = askelia + abs(et);
        robottiX = robottiX + et;
        kolikoita++;
        //auto it = next(kolikotX.begin(), robottiIndeksi); 
        kolikotX.erase(robottiX+(et*-1));
        
        if (kolikotX.size() == 1) {
            break;
        }
    }
    std::cout << askelia << " " << kolikoita << "\n";
    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: TIME LIMIT EXCEEDED

Test 16

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 2

Verdict: ACCEPTED

Test 18

Group: 2

Verdict: TIME LIMIT EXCEEDED

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