Robotti

warning: method `tulosta_huoneet` is never used
   --> input/code.rs:131:8
    |
26  | impl Rakennus {
    | ------------- method in this implementation
...
131 |     fn tulosta_huoneet(&self) {
    |        ^^^^^^^^^^^^^^^
    |
    = note: `#[warn(dead_code)]` on by default

warning: 1 warning emitted

use std::vec::Vec;

fn main() {
    let mut stdin_lines = std::io::stdin().lines();

    // Luetaan vain kartta huoneista, ei tarvita ensimmäistä riviä, joten skipataan se.
    let _ = stdin_lines.next();
    let rivi = stdin_lines.next().unwrap().unwrap();
    let mut rakennus = Rakennus::new(rivi);

    let (askeleet, kolikot) = rakennus.pelaa();
    println!("{} {}", askeleet, kolikot);
}

struct Rakennus {
    // Kartta rakennuksen huoneista
    huoneet: Vec<Huone>,

    // Indeksi huoneet-vektoriin, jossa robotti sijaitsee
    robotti_indeksi: usize,

    keratyt_kolikot: u32,
    askeleet: u32,
}

impl Rakennus {
    fn new(syote: String) -> Self {
        Self {
            huoneet: syote
                .chars()
                .map(|merkki| Huone::from_char(merkki))
                .collect(),
            robotti_indeksi: syote.find("R").unwrap(),
            keratyt_kolikot: 0,
            askeleet: 0,
        }
    }

    // Pelaa peliä. Palauttaa tuplen, jossa ensimmäinen askelten määrä ja toinen kerättyjen
    // kolikoiden määrä
    fn pelaa(&mut self) -> (u32, u32) {
        // self.tulosta_huoneet();
        self.pelaa_inner();
        (self.askeleet, self.keratyt_kolikot)
    }

    // Funktio, joka pelaa peliä yksi kolikon keräys kerrallaan
    fn pelaa_inner(&mut self) {
        loop {
            let lahin_kolikko = match self.etsi_lahin_kolikko() {
                Some(kolikko) => kolikko,
                None => {
                    return;
                }
            };

            self.askeleet += (self.robotti_indeksi as i32 - lahin_kolikko as i32).abs() as u32;
            self.keratyt_kolikot += 1;
            self.robotti_indeksi = lahin_kolikko;
            self.huoneet[lahin_kolikko] = Huone::Tyhja;
            // self.tulosta_huoneet();
        }
    }

    fn etsi_lahin_kolikko(&self) -> Option<usize> {
        let kolikko_vasemmalla = self.etsi_kolikko_vasemmalla();
        let kolikko_oikealla = self.etsi_kolikko_oikealla();
        if let None = kolikko_vasemmalla {
            return kolikko_oikealla;
        }
        if let None = kolikko_oikealla {
            return kolikko_vasemmalla;
        }

        let kolikko_vasemmalla = kolikko_vasemmalla.unwrap();
        let kolikko_oikealla = kolikko_oikealla.unwrap();

        let etaisyys_vasemmalle = self.robotti_indeksi - kolikko_vasemmalla;
        let etaisyys_oikealle = kolikko_oikealla - self.robotti_indeksi;

        if etaisyys_vasemmalle > etaisyys_oikealle {
            Some(kolikko_oikealla)
        } else if etaisyys_oikealle > etaisyys_vasemmalle {
            Some(kolikko_vasemmalla)
        } else {
            None
        }
    }

    fn etsi_kolikko_oikealla(&self) -> Option<usize> {
        // let mut i: usize = self.robotti_indeksi;
        // loop {
        //     if i >= self.huoneet.len() - 1 {
        //         return None;
        //     }
        //     i += 1;
        //     if self.huoneet[i] == Huone::Kolikko {
        //         return Some(i);
        //     }
        // }
        match self.huoneet[self.robotti_indeksi..self.huoneet.len()]
            .iter()
            .position(|huone| *huone == Huone::Kolikko)
        {
            Some(p) => Some(p + self.robotti_indeksi),
            None => None,
        }
    }

    fn etsi_kolikko_vasemmalla(&self) -> Option<usize> {
        // let mut i: usize = self.robotti_indeksi;
        // loop {
        //     if i <= 0 {
        //         return None;
        //     }
        //     i -= 1;
        //     if self.huoneet[i] == Huone::Kolikko {
        //         return Some(i);
        //     }
        // }
        match self.huoneet[0..=self.robotti_indeksi]
            .iter()
            .rev()
            .position(|huone| *huone == Huone::Kolikko)
        {
            Some(p) => Some(self.robotti_indeksi - p),
            None => None,
        }
    }

    fn tulosta_huoneet(&self) {
        let mut string = self
            .huoneet
            .iter()
            .map(|huone| match huone {
                Huone::Tyhja => ".",
                Huone::Kolikko => "*",
            })
            .collect::<String>();
        string.replace_range(self.robotti_indeksi..self.robotti_indeksi + 1, "R");
        println!("Huoneet: {}", string);
    }
}

#[derive(PartialEq, Eq)]
enum Huone {
    Tyhja,
    Kolikko,
}

impl Huone {
    fn from_char(character: char) -> Self {
        match character {
            '*' => Huone::Kolikko,
            '.' => Huone::Tyhja,
            'R' => Huone::Tyhja,
            _ => panic!("Viallinen merkki stdin:ssa"),
        }
    }
}

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