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() { 
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(NULL);
    int n;
    std::cin >> n;
 
    std::set<int> kolikotX;
    int robottiX = 0;
 
    char inp;
    int indx = 0;
    
    while (indx<n) {
        std::cin >> inp;
        indx++;
        if (inp == '*'){
            kolikotX.insert(indx);
        } else if (inp == 'R'){
            robottiX = indx;
            kolikotX.insert(indx);
            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