Robotti

input/code.cpp: In function 'int countCoins(const string&)':
input/code.cpp:35:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   35 |     for (int i=0; i<series.size(); i++) {
      |                   ~^~~~~~~~~~~~~~
input/code.cpp: In function 'std::vector<int> getCharIdxs(const string&, char)':
input/code.cpp:44:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   44 |     for (int i=0; i<series.size(); i++) {
      |                   ~^~~~~~~~~~~~~~
input/code.cpp: In function 'std::vector<int> getStepsClosestCoinIdx(const string&, int)':
input/code.cpp:59:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |     for (int i=1; i<coin...

#include <iostream>
#include <string>
#include <vector>
 
int getDistance(int idx1, int idx2);
 
int countCoins(const std::string& series); // carrots taste good ¯\_(ツ)_/¯
 
std::vector<int> getCharIdxs(const std::string& series, const char carrot); 
 
std::vector<int> getStepsClosestCoinIdx(const std::string& series, int robotPos);
 
std::vector<int> getStepsCoins(std::string& roomSeries);
 
int main() {
    int roomAmount;
    std::string roomSeries;
 
    std::cin >> roomAmount;
    std:: cin >> roomSeries;
 
    std::vector<int> stepsCoins = getStepsCoins(roomSeries);
 
    std::cout << stepsCoins.at(0) << " " << stepsCoins.at(1);
 
    return 0;
}
 
int getDistance(int idx1, int idx2) {
    return abs(idx2 - idx1);
}
 
int countCoins(const std::string& series) {
    int count = 0;
    for (int i=0; i<series.size(); i++) {
        if (series[i] == '*') count++;
    }
    return count;
}
 
std::vector<int> getCharIdxs(const std::string& series, const char carrot) {
    std::vector<int> carrotIndexes;
 
    for (int i=0; i<series.size(); i++) {
        if (series[i] == carrot) carrotIndexes.push_back(i);
    }
 
    return carrotIndexes;
}
 
std::vector<int> getStepsClosestCoinIdx(const std::string& series, int robotIdx) {
    int shortestDistance, distance, closestCoinIdx;
    std::vector<int> coinIdxs;
 
    coinIdxs = getCharIdxs(series, '*');
    closestCoinIdx = coinIdxs.at(0);
    shortestDistance = getDistance(robotIdx, closestCoinIdx);
 
    for (int i=1; i<coinIdxs.size(); i++) {
        distance = getDistance(robotIdx, coinIdxs.at(i));
        if (distance < shortestDistance) {
            closestCoinIdx = coinIdxs.at(i);
            shortestDistance = distance;
        } else if (distance == shortestDistance) {
            return {shortestDistance, closestCoinIdx, i};
        }
    }
 
    return {shortestDistance, closestCoinIdx};
}
 
std::vector<int> getStepsCoins(std::string& roomSeries) {
    int steps=0, gatheredCoins=0, robotIdx, coinIdx;
 
    int coinsTotalAmount = countCoins(roomSeries);
 
    robotIdx = getCharIdxs(roomSeries, 'R').at(0);
 
    std::vector<int> stepsCoinIdx;
 
    for (int i=0; i<coinsTotalAmount; i++) {
        stepsCoinIdx = getStepsClosestCoinIdx(roomSeries, robotIdx);
 
        if (stepsCoinIdx.size() > 2) break;
 
        steps += stepsCoinIdx.at(0);
        coinIdx = stepsCoinIdx.at(1);
        gatheredCoins++;
 
        if (gatheredCoins >= coinsTotalAmount) break;
 
        roomSeries[coinIdx] = 'R';
        roomSeries[robotIdx] = '.';
        robotIdx = coinIdx;
    } 
 
    return {steps, gatheredCoins};
}
 
 
 
// *..*...R.*..*.

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: 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: TIME LIMIT EXCEEDED

Test 23

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 2

Verdict: TIME LIMIT EXCEEDED