Robotti

warning: field `kartan_pituus` is never read
  --> input/code.rs:17:5
   |
15 | struct Rakennus {
   |        -------- field in this struct
16 |     kolikot: Vec<Kolikko>,
17 |     kartan_pituus: usize,
   |     ^^^^^^^^^^^^^
   |
   = note: `#[warn(dead_code)]` on by default

warning: variants `Tyhja` and `Kolikko` are never constructed
   --> input/code.rs:195:5
    |
194 | enum Huone {
    |      ----- variants in this enum
195 |     Tyhja,
    |     ^^^^^
196 |     Kolikko,
    |     ^^^^^^^

warning: associated function `from_char` is never used
   --> input/code.rs:205:8
    |
204 | impl Huone {
    | ---------- associated function in this implementation
205 |     fn from_char(character: char) -> Self {
    |        ^^^^^^^^^

warning: 3 warnings 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 {
    kolikot: Vec<Kolikko>,
    kartan_pituus: usize,
    // Indeksi huoneet-vektoriin, jossa robotti sijaitsee
    robotti_indeksi: usize,

    // Edellinen indeksi kolikot-vektoriin, jossa kolikko on sijainnut.
    edellinen_kolikkoindeksi: usize,

    keratyt_kolikot: u32,
    askeleet: u32,
}

impl Rakennus {
    fn new(syote: String) -> Self {
        Self {
            kolikot: {
                let mut kolikot = vec![];
                for (indeksi, merkki) in syote.chars().enumerate() {
                    if merkki == '*' {
                        kolikot.push(Kolikko {
                            keratty: false,
                            indeksi,
                        });
                    }
                }
                kolikot
            },
            robotti_indeksi: syote.find("R").unwrap(),
            edellinen_kolikkoindeksi: 0,
            keratyt_kolikot: 0,
            askeleet: 0,
            kartan_pituus: syote.len(),
        }
    }

    // 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) {
        {
            // println!("Robo idx: {}", self.robotti_indeksi);
            let lahin_kolikko = match self.etsi_ensimmainen_kolikko() {
                Some(kolikko) => kolikko,
                None => {
                    return;
                }
            };

            self.askeleet += (self.robotti_indeksi as i32
                - self.kolikot[lahin_kolikko].indeksi as i32)
                .abs() as u32;
            self.keratyt_kolikot += 1;
            self.robotti_indeksi = self.kolikot[lahin_kolikko].indeksi;
            self.kolikot[lahin_kolikko].keratty = true;
            self.edellinen_kolikkoindeksi = lahin_kolikko;
            // println!("Robo idx: {}", self.robotti_indeksi);
        }
        // self.tulosta_huoneet();
        loop {
            let lahin_kolikko = match self.etsi_lahin_kolikko() {
                Some(kolikko) => kolikko,
                None => {
                    return;
                }
            };

            self.askeleet += (self.robotti_indeksi as i32
                - self.kolikot[lahin_kolikko].indeksi as i32)
                .abs() as u32;
            self.keratyt_kolikot += 1;
            self.robotti_indeksi = self.kolikot[lahin_kolikko].indeksi;
            self.kolikot[lahin_kolikko].keratty = true;
            self.edellinen_kolikkoindeksi = lahin_kolikko;
            // println!("Robo idx: {}", self.robotti_indeksi);
            // self.tulosta_huoneet();
        }
    }

    fn etsi_ensimmainen_kolikko(&self) -> Option<usize> {
        let kolikko_oikealla_indeksi = self
            .kolikot
            .iter()
            .position(|kolikko| kolikko.indeksi > self.robotti_indeksi);
        if let None = kolikko_oikealla_indeksi {
            if self.kolikot.len() > 0 {
                return Some(self.kolikot.len() - 1);
            } else {
                return None;
            }
        }
        let kolikko_oikealla_indeksi = kolikko_oikealla_indeksi.unwrap();
        if kolikko_oikealla_indeksi == 0 {
            return Some(kolikko_oikealla_indeksi);
        }
        let kolikko_vasemmalla_indeksi = kolikko_oikealla_indeksi - 1;

        let etaisyys_oikealle =
            self.kolikot[kolikko_oikealla_indeksi].indeksi - self.robotti_indeksi;
        let etaisyys_vasemmalle =
            self.robotti_indeksi - self.kolikot[kolikko_vasemmalla_indeksi].indeksi;

        if etaisyys_vasemmalle > etaisyys_oikealle {
            Some(kolikko_oikealla_indeksi)
        } else if etaisyys_oikealle > etaisyys_vasemmalle {
            Some(kolikko_vasemmalla_indeksi)
        } else {
            None
        }
    }

    // Etsii lahimman kolikon ja jos löytyy, palauttaa indeksin kolikot-vektoriin
    fn etsi_lahin_kolikko(&self) -> Option<usize> {
        let kolikko_indeksi_vasemmalla = self.etsi_kolikko_vasemmalla();
        let kolikko_indeksi_oikealla = self.etsi_kolikko_oikealla();
        if let None = kolikko_indeksi_vasemmalla {
            return kolikko_indeksi_oikealla;
        }
        if let None = kolikko_indeksi_oikealla {
            return kolikko_indeksi_vasemmalla;
        }

        let kolikko_vasemmalla = &self.kolikot[kolikko_indeksi_vasemmalla.unwrap()];
        let kolikko_oikealla = &self.kolikot[kolikko_indeksi_oikealla.unwrap()];

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

        if etaisyys_vasemmalle > etaisyys_oikealle {
            Some(kolikko_indeksi_oikealla.unwrap())
        } else if etaisyys_oikealle > etaisyys_vasemmalle {
            Some(kolikko_indeksi_vasemmalla.unwrap())
        } else {
            None
        }
    }

    fn etsi_kolikko_oikealla(&self) -> Option<usize> {
        match self.kolikot[self.edellinen_kolikkoindeksi..self.kolikot.len()]
            .iter()
            .position(|kolikko| !kolikko.keratty)
        {
            Some(pos) => Some(pos + self.edellinen_kolikkoindeksi),
            None => None,
        }
    }

    fn etsi_kolikko_vasemmalla(&self) -> Option<usize> {
        match self.kolikot[0..=self.edellinen_kolikkoindeksi]
            .iter()
            .rev()
            .position(|kolikko| !kolikko.keratty)
        {
            Some(pos) => Some(self.edellinen_kolikkoindeksi - pos),
            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,
}

struct Kolikko {
    keratty: bool,
    indeksi: usize,
}

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