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