Robotti

input/code.cpp: In function 'std::vector<int> CollectCoins(int, std::string, std::vector<int>)':
input/code.cpp:22:22: warning: comparison of integer expressions of different signedness: 'std::size_t' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   22 |         if (CurRobot != Rooms+1) {
      |             ~~~~~~~~~^~~~~~~~~~

#include <cstddef>
#include <iostream>
#include <string>
#include <vector>
 
std::vector<int> CollectCoins(int Rooms, std::string Layout, std::vector<int> Output) {
	std::size_t CurRobot = Layout.find("R");
 
	// Distance to the first coin left
	std::size_t CoinLeft = -1;
	std::size_t DistanceLeft = -1;
	if (CurRobot != 0) {
		CoinLeft = Layout.find_last_of("*", CurRobot);
		//std::cout << CoinLeft << "\n";
		DistanceLeft = CurRobot - CoinLeft;
	}
	//std::cout << DistanceLeft << "\n";
 
	// Distance to the first coin right
	std::size_t CoinRight = -1;
	std::size_t DistanceRight = -1;
	if (CurRobot != Rooms+1) {
		CoinRight = Layout.find_first_of("*", CurRobot);
		//std::cout << CoinRight << "\n";
		DistanceRight = CoinRight - CurRobot;
	}
	//std::cout << DistanceRight << "\n";
 
	if (DistanceLeft==DistanceRight || ((CoinLeft == (size_t)-1) && (CoinRight == (size_t)-1))) {
		return Output;
	}
 
	Layout[CurRobot] = '.';
	Output[1]++;
 
	if (DistanceLeft<DistanceRight) {
		Output[0] += DistanceLeft;
		Layout[CoinLeft] = 'R';
	} else {
		Output[0] += DistanceRight;
		Layout[CoinRight] = 'R';
	}
	//std::cout << Layout << "\n";
 
	return CollectCoins(Rooms, Layout, Output);
}
 
int main () {
	// Input:
	// n rooms
	// Building layout
	// 	e.g.
	// 	20
	// 	**.*......*.R*...*..
	// * = coin; . = empty; R = robot
	int Rooms;
	std::string Layout;
	std::cin >> Rooms;
	std::cin >> Layout;
 
	// GOAL: Find the max number of coins the robot can collect by always going to
	// the closest coin
	std::vector<int> Output = {0,0};
	Output = CollectCoins(Rooms, Layout, Output);
	int Steps = Output[0];
	int Coins = Output[1];
 
	std::cout << Steps << " " << Coins << "\n";
}
 

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: RUNTIME ERROR

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: RUNTIME ERROR

Test 19

Group: 2

Verdict: RUNTIME ERROR

Test 20

Group: 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: ACCEPTED

Test 22

Group: 2

Verdict: RUNTIME ERROR

Test 23

Group: 2

Verdict: RUNTIME ERROR

Test 24

Group: 2

Verdict: RUNTIME ERROR