From 55655442a1acb22e348fa7b3ebd712c13f1736ce Mon Sep 17 00:00:00 2001 From: Jur van den Berg Date: Mon, 5 Dec 2022 20:21:06 +0100 Subject: [PATCH] day 5 --- inputs/day5.txt | 513 +++++++++++++++++++++++++++++++++++++++++++++ src/bin/day05_1.rs | 7 + src/bin/day05_2.rs | 7 + src/day05.rs | 134 ++++++++++++ src/lib.rs | 1 + 5 files changed, 662 insertions(+) create mode 100644 inputs/day5.txt create mode 100644 src/bin/day05_1.rs create mode 100644 src/bin/day05_2.rs create mode 100644 src/day05.rs diff --git a/inputs/day5.txt b/inputs/day5.txt new file mode 100644 index 0000000..cf16741 --- /dev/null +++ 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 diff --git a/src/bin/day05_1.rs b/src/bin/day05_1.rs new file mode 100644 index 0000000..9f93792 --- /dev/null +++ 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)); +} diff --git a/src/bin/day05_2.rs b/src/bin/day05_2.rs new file mode 100644 index 0000000..a14c9ea --- /dev/null +++ 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)); +} diff --git a/src/day05.rs b/src/day05.rs new file mode 100644 index 0000000..5bfd5a7 --- /dev/null +++ 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::>() + .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::>() + .join("") +} + +#[derive(Debug)] +struct Move { + count: u32, + from: u32, + to: u32, +} + +fn parse_crates(input: &str) -> IResult<&str, (Vec>, Vec, Vec)> { + 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> = (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> { + 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>>> { + separated_list1(newline, parse_crate_line)(input) +} + +fn parse_crate_line(input: &str) -> IResult<&str, Vec>> { + 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"); + } +} diff --git a/src/lib.rs b/src/lib.rs index 3d7adfa..59631f5 100644 --- 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;