day 5

2022-12-05 20:21:06 +01:00
parent 96053f5345
commit 55655442a1
5 changed files with 662 additions and 0 deletions
--- a/inputs/day5.txt
+++ b/inputs/day5.txt
@@ -0,0 +1,513 @@
+                [M]     [V]     [L]
+[G]             [V] [C] [G]     [D]
+[J]             [Q] [W] [Z] [C] [J]
+[W]         [W] [G] [V] [D] [G] [C]
+[R]     [G] [N] [B] [D] [C] [M] [W]
+[F] [M] [H] [C] [S] [T] [N] [N] [N]
+[T] [W] [N] [R] [F] [R] [B] [J] [P]
+[Z] [G] [J] [J] [W] [S] [H] [S] [G]
+ 1   2   3   4   5   6   7   8   9 
+
+move 1 from 5 to 2
+move 7 from 7 to 1
+move 1 from 1 to 7
+move 1 from 4 to 1
+move 7 from 9 to 1
+move 1 from 3 to 7
+move 4 from 5 to 4
+move 6 from 4 to 9
+move 2 from 7 to 6
+move 6 from 8 to 2
+move 2 from 4 to 5
+move 2 from 3 to 7
+move 11 from 1 to 4
+move 6 from 6 to 1
+move 3 from 5 to 3
+move 5 from 9 to 8
+move 1 from 2 to 3
+move 2 from 7 to 9
+move 7 from 1 to 2
+move 1 from 5 to 3
+move 1 from 5 to 3
+move 5 from 8 to 5
+move 3 from 5 to 4
+move 1 from 1 to 7
+move 1 from 3 to 8
+move 2 from 6 to 3
+move 3 from 3 to 4
+move 1 from 6 to 2
+move 5 from 4 to 2
+move 2 from 5 to 3
+move 2 from 7 to 1
+move 1 from 8 to 1
+move 7 from 1 to 7
+move 4 from 4 to 2
+move 7 from 4 to 1
+move 10 from 1 to 5
+move 10 from 5 to 2
+move 11 from 2 to 3
+move 1 from 1 to 6
+move 1 from 4 to 7
+move 4 from 7 to 1
+move 6 from 2 to 5
+move 2 from 1 to 3
+move 1 from 9 to 5
+move 2 from 9 to 6
+move 1 from 6 to 1
+move 3 from 5 to 4
+move 20 from 3 to 9
+move 3 from 7 to 1
+move 3 from 5 to 2
+move 3 from 4 to 8
+move 3 from 1 to 3
+move 3 from 1 to 2
+move 2 from 6 to 1
+move 10 from 9 to 6
+move 6 from 6 to 7
+move 4 from 6 to 3
+move 11 from 2 to 6
+move 1 from 8 to 9
+move 13 from 2 to 3
+move 1 from 1 to 9
+move 1 from 9 to 4
+move 1 from 8 to 2
+move 1 from 8 to 2
+move 4 from 7 to 8
+move 8 from 6 to 9
+move 3 from 2 to 3
+move 3 from 8 to 4
+move 11 from 9 to 2
+move 7 from 9 to 6
+move 1 from 1 to 5
+move 4 from 4 to 9
+move 21 from 3 to 1
+move 1 from 3 to 9
+move 7 from 6 to 3
+move 6 from 1 to 2
+move 13 from 1 to 5
+move 2 from 1 to 2
+move 3 from 9 to 3
+move 2 from 2 to 3
+move 2 from 6 to 4
+move 3 from 3 to 5
+move 13 from 5 to 2
+move 5 from 3 to 4
+move 2 from 7 to 9
+move 2 from 4 to 2
+move 1 from 3 to 8
+move 1 from 6 to 1
+move 4 from 3 to 7
+move 2 from 5 to 7
+move 1 from 7 to 2
+move 1 from 5 to 9
+move 4 from 7 to 8
+move 1 from 1 to 9
+move 6 from 8 to 1
+move 4 from 4 to 8
+move 25 from 2 to 9
+move 1 from 4 to 3
+move 1 from 3 to 7
+move 4 from 8 to 1
+move 1 from 7 to 4
+move 3 from 1 to 6
+move 5 from 2 to 1
+move 1 from 5 to 1
+move 1 from 4 to 1
+move 24 from 9 to 6
+move 9 from 1 to 6
+move 1 from 5 to 6
+move 1 from 1 to 9
+move 1 from 2 to 8
+move 1 from 8 to 1
+move 3 from 1 to 8
+move 36 from 6 to 3
+move 2 from 7 to 3
+move 1 from 2 to 5
+move 1 from 5 to 2
+move 1 from 6 to 2
+move 10 from 3 to 2
+move 3 from 8 to 2
+move 1 from 1 to 7
+move 2 from 2 to 6
+move 10 from 9 to 1
+move 2 from 6 to 4
+move 13 from 3 to 4
+move 8 from 3 to 7
+move 8 from 1 to 2
+move 5 from 3 to 8
+move 3 from 1 to 9
+move 1 from 7 to 1
+move 7 from 4 to 5
+move 1 from 1 to 2
+move 14 from 2 to 6
+move 2 from 7 to 2
+move 8 from 4 to 8
+move 3 from 7 to 9
+move 2 from 9 to 8
+move 2 from 7 to 1
+move 1 from 7 to 8
+move 1 from 6 to 8
+move 1 from 9 to 3
+move 4 from 2 to 7
+move 6 from 6 to 1
+move 3 from 1 to 9
+move 1 from 1 to 7
+move 6 from 5 to 6
+move 1 from 5 to 2
+move 1 from 6 to 8
+move 5 from 7 to 5
+move 1 from 2 to 9
+move 2 from 3 to 4
+move 9 from 8 to 4
+move 8 from 4 to 8
+move 6 from 6 to 7
+move 5 from 6 to 4
+move 7 from 9 to 7
+move 7 from 8 to 7
+move 5 from 8 to 4
+move 3 from 1 to 6
+move 1 from 2 to 7
+move 1 from 1 to 4
+move 4 from 5 to 2
+move 2 from 6 to 9
+move 1 from 3 to 7
+move 1 from 5 to 1
+move 1 from 8 to 9
+move 1 from 6 to 1
+move 1 from 2 to 7
+move 2 from 8 to 1
+move 2 from 1 to 8
+move 3 from 2 to 4
+move 1 from 6 to 1
+move 17 from 4 to 1
+move 3 from 2 to 7
+move 13 from 7 to 8
+move 1 from 2 to 6
+move 14 from 1 to 4
+move 2 from 8 to 5
+move 1 from 9 to 7
+move 2 from 5 to 4
+move 1 from 9 to 3
+move 5 from 1 to 5
+move 3 from 4 to 1
+move 1 from 3 to 2
+move 7 from 4 to 5
+move 9 from 7 to 8
+move 5 from 4 to 2
+move 1 from 1 to 3
+move 1 from 9 to 2
+move 15 from 8 to 6
+move 1 from 3 to 7
+move 11 from 6 to 5
+move 1 from 4 to 8
+move 3 from 1 to 7
+move 5 from 7 to 5
+move 27 from 5 to 1
+move 8 from 8 to 4
+move 1 from 2 to 6
+move 3 from 6 to 1
+move 9 from 1 to 5
+move 5 from 5 to 7
+move 2 from 2 to 1
+move 2 from 5 to 4
+move 6 from 7 to 6
+move 1 from 5 to 2
+move 1 from 7 to 8
+move 4 from 6 to 8
+move 5 from 6 to 3
+move 1 from 7 to 1
+move 5 from 4 to 3
+move 6 from 8 to 2
+move 1 from 7 to 8
+move 2 from 8 to 9
+move 10 from 3 to 5
+move 9 from 5 to 2
+move 3 from 4 to 8
+move 1 from 5 to 7
+move 2 from 9 to 7
+move 2 from 8 to 3
+move 1 from 3 to 8
+move 19 from 1 to 7
+move 4 from 2 to 7
+move 2 from 4 to 3
+move 3 from 3 to 2
+move 2 from 8 to 3
+move 2 from 5 to 8
+move 1 from 2 to 3
+move 2 from 8 to 3
+move 5 from 2 to 5
+move 9 from 7 to 5
+move 13 from 5 to 9
+move 7 from 2 to 6
+move 2 from 6 to 9
+move 1 from 2 to 1
+move 5 from 6 to 7
+move 1 from 5 to 7
+move 6 from 1 to 2
+move 5 from 3 to 6
+move 6 from 7 to 2
+move 3 from 6 to 4
+move 3 from 7 to 4
+move 12 from 7 to 6
+move 5 from 4 to 1
+move 2 from 7 to 4
+move 3 from 4 to 6
+move 16 from 6 to 3
+move 4 from 1 to 4
+move 1 from 1 to 9
+move 3 from 9 to 2
+move 1 from 4 to 6
+move 9 from 3 to 7
+move 2 from 6 to 3
+move 3 from 3 to 9
+move 15 from 2 to 7
+move 19 from 7 to 4
+move 15 from 9 to 2
+move 16 from 2 to 8
+move 6 from 3 to 5
+move 4 from 7 to 5
+move 15 from 8 to 7
+move 19 from 4 to 2
+move 1 from 8 to 3
+move 16 from 2 to 1
+move 9 from 7 to 6
+move 7 from 2 to 8
+move 2 from 2 to 7
+move 1 from 9 to 5
+move 1 from 3 to 4
+move 6 from 1 to 2
+move 8 from 5 to 1
+move 1 from 5 to 1
+move 18 from 1 to 8
+move 7 from 7 to 5
+move 7 from 5 to 3
+move 4 from 3 to 6
+move 13 from 8 to 5
+move 12 from 8 to 1
+move 5 from 1 to 6
+move 15 from 5 to 4
+move 1 from 1 to 6
+move 12 from 6 to 3
+move 8 from 3 to 4
+move 2 from 7 to 3
+move 9 from 3 to 1
+move 5 from 2 to 9
+move 16 from 4 to 3
+move 10 from 1 to 3
+move 2 from 1 to 5
+move 1 from 3 to 1
+move 5 from 6 to 1
+move 4 from 9 to 3
+move 1 from 2 to 8
+move 1 from 8 to 1
+move 1 from 9 to 8
+move 2 from 5 to 9
+move 9 from 4 to 1
+move 3 from 1 to 3
+move 2 from 6 to 8
+move 3 from 8 to 5
+move 2 from 1 to 5
+move 2 from 9 to 8
+move 1 from 8 to 6
+move 2 from 5 to 3
+move 19 from 3 to 1
+move 2 from 4 to 2
+move 1 from 5 to 6
+move 2 from 2 to 3
+move 1 from 8 to 6
+move 8 from 3 to 9
+move 6 from 3 to 7
+move 2 from 6 to 2
+move 1 from 6 to 1
+move 1 from 1 to 8
+move 1 from 8 to 9
+move 1 from 7 to 3
+move 19 from 1 to 5
+move 21 from 5 to 2
+move 13 from 2 to 6
+move 13 from 1 to 8
+move 7 from 9 to 7
+move 2 from 9 to 2
+move 10 from 8 to 3
+move 1 from 1 to 6
+move 10 from 2 to 4
+move 11 from 3 to 5
+move 8 from 5 to 6
+move 1 from 3 to 7
+move 2 from 8 to 6
+move 2 from 2 to 8
+move 3 from 7 to 6
+move 2 from 8 to 6
+move 1 from 1 to 2
+move 24 from 6 to 5
+move 2 from 3 to 8
+move 1 from 8 to 6
+move 7 from 7 to 9
+move 4 from 6 to 9
+move 1 from 8 to 9
+move 21 from 5 to 9
+move 2 from 7 to 2
+move 1 from 8 to 5
+move 1 from 7 to 3
+move 12 from 9 to 6
+move 6 from 6 to 3
+move 12 from 9 to 4
+move 4 from 5 to 6
+move 13 from 4 to 2
+move 8 from 4 to 8
+move 10 from 6 to 8
+move 11 from 8 to 9
+move 4 from 8 to 4
+move 2 from 4 to 3
+move 8 from 3 to 8
+move 2 from 6 to 8
+move 1 from 3 to 8
+move 6 from 2 to 4
+move 1 from 4 to 8
+move 1 from 9 to 7
+move 13 from 8 to 4
+move 1 from 7 to 1
+move 1 from 1 to 4
+move 8 from 4 to 7
+move 3 from 5 to 7
+move 19 from 9 to 7
+move 3 from 2 to 7
+move 1 from 8 to 2
+move 13 from 7 to 6
+move 1 from 2 to 4
+move 4 from 6 to 2
+move 1 from 8 to 3
+move 7 from 6 to 8
+move 1 from 6 to 2
+move 1 from 2 to 7
+move 9 from 2 to 3
+move 1 from 6 to 2
+move 21 from 7 to 5
+move 9 from 5 to 3
+move 19 from 3 to 9
+move 5 from 8 to 5
+move 2 from 2 to 1
+move 2 from 1 to 8
+move 6 from 4 to 5
+move 3 from 8 to 7
+move 15 from 9 to 2
+move 2 from 2 to 5
+move 3 from 9 to 6
+move 5 from 4 to 5
+move 11 from 2 to 6
+move 1 from 8 to 6
+move 1 from 9 to 5
+move 1 from 7 to 3
+move 6 from 5 to 6
+move 1 from 4 to 6
+move 1 from 3 to 4
+move 13 from 5 to 2
+move 16 from 6 to 9
+move 4 from 4 to 5
+move 2 from 6 to 2
+move 2 from 6 to 4
+move 2 from 4 to 5
+move 2 from 7 to 8
+move 2 from 6 to 3
+move 2 from 5 to 8
+move 14 from 5 to 7
+move 4 from 8 to 1
+move 4 from 1 to 6
+move 1 from 3 to 9
+move 1 from 6 to 1
+move 2 from 7 to 3
+move 2 from 3 to 7
+move 2 from 5 to 2
+move 9 from 9 to 2
+move 13 from 7 to 3
+move 12 from 3 to 9
+move 2 from 6 to 8
+move 14 from 2 to 9
+move 2 from 8 to 9
+move 10 from 2 to 1
+move 1 from 7 to 4
+move 2 from 3 to 8
+move 4 from 2 to 1
+move 1 from 8 to 3
+move 1 from 2 to 6
+move 1 from 8 to 3
+move 4 from 9 to 4
+move 1 from 3 to 5
+move 1 from 5 to 1
+move 1 from 3 to 9
+move 12 from 1 to 8
+move 10 from 8 to 5
+move 7 from 5 to 6
+move 1 from 1 to 9
+move 3 from 5 to 1
+move 1 from 1 to 3
+move 16 from 9 to 7
+move 4 from 4 to 3
+move 1 from 4 to 9
+move 15 from 7 to 8
+move 15 from 9 to 1
+move 8 from 1 to 6
+move 1 from 9 to 3
+move 17 from 6 to 2
+move 1 from 9 to 1
+move 15 from 2 to 7
+move 14 from 8 to 9
+move 12 from 7 to 9
+move 12 from 9 to 3
+move 3 from 7 to 9
+move 1 from 7 to 4
+move 7 from 9 to 6
+move 1 from 4 to 6
+move 11 from 9 to 6
+move 2 from 1 to 2
+move 18 from 6 to 4
+move 4 from 2 to 7
+move 2 from 7 to 3
+move 2 from 7 to 8
+move 4 from 1 to 5
+move 1 from 9 to 2
+move 2 from 5 to 4
+move 5 from 1 to 3
+move 2 from 3 to 7
+move 2 from 3 to 9
+move 1 from 6 to 7
+move 1 from 2 to 9
+move 2 from 8 to 1
+move 3 from 1 to 3
+move 2 from 5 to 8
+move 2 from 3 to 5
+move 1 from 5 to 2
+move 1 from 1 to 3
+move 1 from 9 to 2
+move 1 from 9 to 1
+move 3 from 7 to 6
+move 1 from 1 to 9
+move 2 from 8 to 9
+move 1 from 2 to 3
+move 2 from 8 to 2
+move 2 from 6 to 5
+move 1 from 8 to 5
+move 3 from 2 to 5
+move 3 from 4 to 8
+move 1 from 8 to 2
+move 3 from 9 to 7
+move 3 from 7 to 1
+move 1 from 9 to 6
+move 3 from 1 to 2
+move 2 from 8 to 7
+move 2 from 7 to 9
+move 2 from 6 to 5
+move 3 from 5 to 3
+move 1 from 2 to 5
+move 3 from 2 to 7
+move 2 from 5 to 6
+move 15 from 4 to 9
+move 1 from 3 to 1
+move 25 from 3 to 4
+move 3 from 7 to 3
+move 5 from 9 to 5
+move 10 from 9 to 5
+move 9 from 5 to 1
+move 5 from 5 to 2
+move 1 from 6 to 7
+move 5 from 5 to 8
--- a/src/bin/day05_1.rs
+++ b/src/bin/day05_1.rs
@@ -0,0 +1,7 @@
+use aoc2022::day05::process_part_1;
+use std::fs;
+
+fn main() {
+    let file = fs::read_to_string("./inputs/day5.txt").unwrap();
+    println!("{}", process_part_1(&file));
+}
--- a/src/bin/day05_2.rs
+++ b/src/bin/day05_2.rs
@@ -0,0 +1,7 @@
+use aoc2022::day05::process_part_2;
+use std::fs;
+
+fn main() {
+    let file = fs::read_to_string("./inputs/day5.txt").unwrap();
+    println!("{}", process_part_2(&file));
+}
--- a/src/day05.rs
+++ b/src/day05.rs
@@ -0,0 +1,134 @@
+use nom::branch::alt;
+use nom::bytes::complete::tag;
+use nom::character::complete;
+use nom::character::complete::{alpha1, newline};
+use nom::multi::{many1, separated_list1};
+use nom::sequence::delimited;
+use nom::IResult;
+
+pub fn process_part_1(input: &str) -> String {
+    let (_, (mut stacks, _, moves)) = parse_crates(input).unwrap();
+
+    for m in moves {
+        for _ in 0..m.count {
+            let c = stacks[(m.from - 1) as usize]
+                .pop()
+                .expect("expected a crate");
+            stacks[(m.to - 1) as usize].push(c);
+        }
+    }
+
+    stacks
+        .iter_mut()
+        .map(|stack| stack.pop().unwrap_or(""))
+        .collect::<Vec<&str>>()
+        .join("")
+}
+
+pub fn process_part_2(input: &str) -> String {
+    let (_, (mut stacks, _, moves)) = parse_crates(input).unwrap();
+
+    for m in moves {
+        let from_stack = &mut stacks[(m.from - 1) as usize];
+        let mut boxes: Vec<&str> = from_stack
+            .drain((from_stack.len() - m.count as usize)..)
+            .collect();
+        stacks[(m.to - 1) as usize].append(&mut boxes);
+    }
+
+    stacks
+        .iter_mut()
+        .map(|stack| stack.pop().unwrap_or(""))
+        .collect::<Vec<&str>>()
+        .join("")
+}
+
+#[derive(Debug)]
+struct Move {
+    count: u32,
+    from: u32,
+    to: u32,
+}
+
+fn parse_crates(input: &str) -> IResult<&str, (Vec<Vec<&str>>, Vec<u32>, Vec<Move>)> {
+    let (input, crates_transposed) = parse_crate_stacks(input)?;
+    let (input, _) = newline(input)?;
+    let (input, crate_numbers) = separated_list1(tag(" "), parse_crate_name)(input)?;
+    let (input, _) = many1(newline)(input)?; // newlines be gone
+
+    // [
+    //   [ A, B, C ],
+    //   [ D, E, _ ],
+    // ] => [ [D, A], [E, B], [C]
+    // (its transposed for pop() and push() magic :))
+    // it also unwraps all `None`'s from the lists
+    let crates: Vec<Vec<&str>> = (0..crates_transposed[0].len())
+        .map(|i| {
+            crates_transposed
+                .iter()
+                .map(|items| items[i])
+                .rev()
+                .filter_map(|item| item)
+                .collect()
+        })
+        .collect();
+
+    let (input, moves) = parse_moves(input)?;
+    Ok((input, (crates, crate_numbers, moves)))
+}
+
+fn parse_moves(input: &str) -> IResult<&str, Vec<Move>> {
+    separated_list1(newline, parse_move)(input)
+}
+
+fn parse_move(input: &str) -> IResult<&str, Move> {
+    let (input, _) = tag("move ")(input)?;
+    let (input, count) = complete::u32(input)?;
+    let (input, _) = tag(" from ")(input)?;
+    let (input, from) = complete::u32(input)?;
+    let (input, _) = tag(" to ")(input)?;
+    let (input, to) = complete::u32(input)?;
+    Ok((input, Move { count, from, to }))
+}
+
+fn parse_crate_name(input: &str) -> IResult<&str, u32> {
+    delimited(tag(" "), complete::u32, tag(" "))(input)
+}
+
+fn parse_crate_stacks(input: &str) -> IResult<&str, Vec<Vec<Option<&str>>>> {
+    separated_list1(newline, parse_crate_line)(input)
+}
+
+fn parse_crate_line(input: &str) -> IResult<&str, Vec<Option<&str>>> {
+    separated_list1(tag(" "), parse_crate_single)(input)
+}
+
+fn parse_crate_single(input: &str) -> IResult<&str, Option<&str>> {
+    let (input, maybe_crate) = alt((tag("   "), delimited(tag("["), alpha1, tag("]"))))(input)?;
+    match maybe_crate {
+        "   " => Ok((input, None)),
+        _ => Ok((input, Some(maybe_crate))),
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    // Input slightly mangled like this due to rustfmt/intellijfmt stripping trailing spaces
+    const INPUT: &str = "    [D]    \n[N] [C]    \n[Z] [M] [P]\n 1   2   3 \n
+move 1 from 2 to 1
+move 3 from 1 to 3
+move 2 from 2 to 1
+move 1 from 1 to 2";
+
+    #[test]
+    fn day1() {
+        assert_eq!(process_part_1(INPUT), "CMZ");
+    }
+
+    #[test]
+    fn day2() {
+        assert_eq!(process_part_2(INPUT), "MCD");
+    }
+}
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -5,3 +5,4 @@ pub mod day01;
 pub mod day02;
 pub mod day03;
 pub mod day04;
+pub mod day05;