Added day 12 code for part 1 and 2

Signed-off-by: Louis Vallat <louis@louis-vallat.xyz>
This commit is contained in:
Louis Vallat 2021-12-18 00:16:22 +01:00
parent e1d138c3ca
commit b71fba9b61
No known key found for this signature in database
GPG Key ID: 0C87282F76E61283
4 changed files with 107 additions and 0 deletions

View File

@ -58,3 +58,8 @@ day-11:
stage: build
script:
- cd day11; cargo run --release ./input
day-12:
stage: build
script:
- cd day12; cargo run --release ./input

8
day12/Cargo.toml Normal file
View File

@ -0,0 +1,8 @@
[package]
name = "day12"
version = "0.1.0"
edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]

25
day12/input Normal file
View File

@ -0,0 +1,25 @@
bm-XY
ol-JS
bm-im
RD-ol
bm-QI
JS-ja
im-gq
end-im
ja-ol
JS-gq
bm-AF
RD-start
RD-ja
start-ol
cj-bm
start-JS
AF-ol
end-QI
QI-gq
ja-gq
end-AF
im-QI
bm-gq
ja-QI
gq-RD

69
day12/src/main.rs Normal file
View File

@ -0,0 +1,69 @@
use std::{fs, env, collections::HashMap};
fn read_input(path: &str) -> String {
return fs::read_to_string(path).expect("Cannot read file.");
}
fn parse_input(s: &str) -> HashMap<String, Vec<String>> {
let mut m: HashMap<String, Vec<String>> = HashMap::new();
for l in s.lines() {
let t = l.split_once("-").unwrap();
let mut i = m.get(t.0).unwrap_or(&vec![]).to_owned();
let mut j = m.get(t.1).unwrap_or(&vec![]).to_owned();
i.push(t.1.to_string());
j.push(t.0.to_string());
m.insert(t.0.to_string(), i);
m.insert(t.1.to_string(), j);
}
return m;
}
fn is_small_cave(c: &str) -> bool {
if c == "start" || c == "end" { return false; }
return c.chars().fold(true, |acc, x| acc && x.is_lowercase());
}
fn are_all_under_x(p: &(Vec<String>, HashMap<String, u32>), x: u32) -> bool {
for s in p.1.iter() {
if s.1 >= &x { return false; }
}
return true;
}
fn build_paths(v: &HashMap<String, Vec<String>>, c: &str, m: u32, p: &(Vec<String>, HashMap<String, u32>)) -> Option<Vec<(Vec<String>, HashMap<String, u32>)>> {
if c == "end" { return Some(vec![(vec!["end".to_owned()], HashMap::new())]); }
let is_smol = is_small_cave(c);
let mut res = vec![];
for s in v.get(c).unwrap().iter() {
let mut path = p.clone();
let count = *path.1.get(c).unwrap_or(&0);
if s == "start" || (is_smol && (count >= m || (m > 1 && count == m - 1 && !are_all_under_x(p, m)))) { continue; }
if is_smol { path.1.insert(c.to_string(), count + 1); }
path.0.push(s.to_string());
if s == "end" { res.push(path); continue; }
let sub = build_paths(v, s, m, &path);
for u in sub.unwrap_or(vec![]).iter() {
if u.0.last().unwrap_or(&"".to_string()) != "end" { continue; }
res.push(u.to_owned());
}
}
return Some(res);
}
fn main() {
let args: Vec<String> = env::args().collect();
for arg in args.iter().skip(1) {
let input = read_input(&arg);
let vec_in = parse_input(&input);
println!("[{}]", &arg);
println!("\t[Part 1] => Answer is '{}'.",
build_paths(&vec_in, "start", 1, &(vec!["start".to_owned()], HashMap::new()))
.unwrap_or(vec![]).len());
println!("\t[Part 2] => Answer is '{}'.",
build_paths(&vec_in, "start", 2, &(vec!["start".to_owned()], HashMap::new()))
.unwrap_or(vec![]).len());
}
}