Robotti

warning: unused variable: `room_count`
  --> input/code.rs:13:9
   |
13 |     let room_count = n.trim().parse::<i32>().unwrap();
   |         ^^^^^^^^^^ help: if this is intentional, prefix it with an underscore: `_room_count`
   |
   = note: `#[warn(unused_variables)]` on by default

warning: 1 warning emitted

use core::panic;
use std::io::stdin;

#[derive(PartialEq, Eq)]
enum Room {
    Empty,
    HasCoin,
}

fn main() {
    let mut n = String::new();
    stdin().read_line(&mut n).unwrap();
    let room_count = n.trim().parse::<i32>().unwrap();
    let mut room_map = String::new();
    stdin().read_line(&mut room_map).unwrap();
    let mut robot_position: isize = -1;
    let rooms = room_map
        .trim()
        .chars()
        .enumerate()
        .map(|(index, room_desc)| match room_desc {
            '*' => Room::HasCoin,
            '.' => Room::Empty,
            'R' => {
                robot_position = index as isize;
                Room::Empty
            }
            _ => panic!(),
        });

    let mut coin_positions = Vec::new();
    for (index, room) in rooms.enumerate() {
        if room == Room::HasCoin {
            coin_positions.push(index as isize);
        }
    }

    if robot_position == -1 {
        panic!()
    }
    let mut num_coins_collected = 0;
    let mut num_steps_taken = 0;
    loop {
        if coin_positions.is_empty() {
            break;
        }
        let mut closest_coin: isize = -1;
        let mut closest_coin_dis: isize = -1;
        let mut should_exit: bool = false;
        for (index, coin_pos) in coin_positions.iter().enumerate() {
            let dis = robot_position.abs_diff(*coin_pos) as isize;
            if closest_coin == -1 || dis < closest_coin_dis {
                closest_coin = index as isize;
                closest_coin_dis = dis;
                should_exit = false;
            } else if dis == closest_coin_dis {
                should_exit = true;
            }
        }
        if should_exit {
            break;
        }
        robot_position = coin_positions[closest_coin as usize];
        coin_positions.swap_remove(closest_coin as usize);
        num_coins_collected += 1;
        num_steps_taken += closest_coin_dis;
    }
    println!("{} {}", num_steps_taken, num_coins_collected);
}

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: ACCEPTED

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