Robotti

warning: unused variable: `len`
  --> input/code.rs:43:9
   |
43 |     let len: usize = line.strip_suffix('\n').unwrap_or(&line).parse().unwrap();
   |         ^^^ help: if this is intentional, prefix it with an underscore: `_len`
   |
   = note: `#[warn(unused_variables)]` on by default

warning: 1 warning emitted

use std::io::stdin;

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
enum RoomType {
    Robot,
    Coin,
    Empty,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)]
enum Direction {
    Left(i32),
    Right(i32),
    Same,
}

impl Direction {
    fn get_inner(&self) -> i32 {
        match self {
            Direction::Left(x) => *x,
            Direction::Right(x) => *x,
            Direction::Same => 0,
        }
    }
}

impl RoomType {
    fn from_char(c: char) -> Self {
        use RoomType::*;
        match c {
            'R' => Robot,
            '*' => Coin,
            '.' => Empty,
            _ => panic!("Should Be One Of The Above"),
        }
    }
}

fn main() {
    let mut line = String::new();

    stdin().read_line(&mut line).unwrap();
    let len: usize = line.strip_suffix('\n').unwrap_or(&line).parse().unwrap();

    line = "".to_string();

    stdin().read_line(&mut line).unwrap();

    let mut arr: Vec<RoomType> = line
        .strip_suffix('\n')
        .unwrap()
        .chars()
        .map(|c| RoomType::from_char(c))
        .collect();

    let mut startpos = arr
        .iter()
        .position(|x| *x == RoomType::Robot)
        .expect("No robot");

    let mut coin_count = 0;
    let mut steps = 0;

    let mut next_coin = find_next_coin(startpos, &arr);

    while next_coin != Direction::Same {
        coin_count += 1;
        steps += next_coin.get_inner();

        match next_coin {
            Direction::Left(x) => startpos -= x as usize,
            Direction::Right(x) => startpos += x as usize,
            Direction::Same => unreachable!(),
        };

        arr[startpos] = RoomType::Empty;

        next_coin = find_next_coin(startpos, &arr)
    }

    println!("{} {}", steps, coin_count);
}

fn find_next_coin(startpos: usize, arr: &Vec<RoomType>) -> Direction {
    let mut rp = startpos.checked_add(1);
    let mut lp = startpos.checked_sub(1);

    let mut distance_to_left_room = 1;
    let mut distance_to_right_room = 1;

    while let Some(room) = get_room(rp, &arr) {
        match room {
            RoomType::Robot => (),
            RoomType::Coin => break,
            RoomType::Empty => (),
        }

        distance_to_right_room += 1;
        rp = match rp {
            Some(x) => x.checked_add(1),
            None => None,
        };
    }

    while let Some(room) = get_room(lp, &arr) {
        match room {
            RoomType::Robot => (),
            RoomType::Coin => break,
            RoomType::Empty => (),
        }

        distance_to_left_room += 1;
        lp = match lp {
            Some(x) => x.checked_sub(1),
            None => None,
        };
    }

    match distance_to_left_room.cmp(&distance_to_right_room) {
        std::cmp::Ordering::Less => Direction::Left(distance_to_left_room),
        std::cmp::Ordering::Equal => Direction::Same,
        std::cmp::Ordering::Greater => Direction::Right(distance_to_right_room),
    }
}

fn get_room(pos: Option<usize>, arr: &Vec<RoomType>) -> Option<RoomType> {
    match pos {
        Some(p) => arr.get(p).or(None).copied(),
        None => None,
    }
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 10 but the index is 18446744073709551615', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 10 but the index is 18446744073709551615', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 1000 but the index is 18446744073709551615', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 1000 but the index is 1000', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 7

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 1000 but the index is 1000', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 1, 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: WRONG ANSWER

Test 11

Group: 1, 2

Verdict: WRONG ANSWER

Test 12

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 10000 but the index is 18446744073709551615', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 13

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 10000 but the index is 10000', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 14

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 10000 but the index is 10000', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 15

Group: 2

Verdict: WRONG ANSWER

Test 16

Group: 2

Verdict: WRONG ANSWER

Test 17

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 200000 but the index is 200000', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 18

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 200000 but the index is 18446744073709551615', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 19

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 200000 but the index is 18446744073709551615', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 22

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 200000 but the index is 18446744073709551615', input/code.rs:76:9
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

Test 23

Group: 2

Verdict: WRONG ANSWER

Test 24

Group: 2

Verdict: WRONG ANSWER