From e8358006ad414c144a6260166f8d3322ca24eb20 Mon Sep 17 00:00:00 2001 From: jacopograndi Date: Wed, 3 Aug 2022 23:21:45 +0200 Subject: day03 --- 2020/day03/src/main.rs | 74 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 74 insertions(+) create mode 100644 2020/day03/src/main.rs (limited to '2020/day03/src/main.rs') diff --git a/2020/day03/src/main.rs b/2020/day03/src/main.rs new file mode 100644 index 0000000..6e74f74 --- /dev/null +++ b/2020/day03/src/main.rs @@ -0,0 +1,74 @@ +fn slide(lines: Vec>, vx: i32, vy: i32) -> i32 { + let mut x: i32 = 0; + let mut y: i32 = 0; + let mut hits: i32 = 0; + loop { + if lines[y as usize][x as usize] == '#' { + hits += 1; + } + x += vx; + y += vy; + if y as usize >= lines.len() { + break; + } + if x as usize >= lines[y as usize].len() { + x -= lines[y as usize].len() as i32; + } + } + hits +} + +fn slide_everywhere(lines: Vec>) -> i32 { + let mut hits: i32 = 1; + for (vx, vy) in vec![(1, 1), (3, 1), (5, 1), (7, 1), (1, 2)] { + hits *= slide(lines.clone(), vx, vy); + } + hits +} + +fn main() { + let lines: Vec> = include_str!("../input0.txt") + .trim() + .split("\n") + .map(|s| s.chars().collect()) + .collect(); + let hits = slide(lines.clone(), 3, 1); + println!("hits: {}", hits); + let hits_everything = slide_everywhere(lines.clone()); + println!("hits on the five paths: {}", hits_everything); +} + +#[cfg(test)] +mod tests { + use crate::*; + + #[test] + fn small_input() { + let lines: Vec> = include_str!("../test0.txt") + .trim() + .split("\n") + .map(|s| s.chars().collect()) + .collect(); + assert_eq!(slide(lines, 3, 1), 7); + } + + #[test] + fn big_input() { + let lines: Vec> = include_str!("../input0.txt") + .trim() + .split("\n") + .map(|s| s.chars().collect()) + .collect(); + assert_eq!(slide(lines, 3, 1), 220); + } + + #[test] + fn small_input_all_slopes() { + let lines: Vec> = include_str!("../test0.txt") + .trim() + .split("\n") + .map(|s| s.chars().collect()) + .collect(); + assert_eq!(slide_everywhere(lines), 336); + } +} -- cgit v1.2.3-54-g00ecf