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"),
}
}
}
