jdkaplan/advent-of-code
Watch
1
0
Fork
You've already forked advent-of-code
 0
Code Issues Activity

Support arbitrary number of stacks

Browse source
This commit is contained in:
Jeremy Kaplan 2022-12-04 23:41:18 -08:00
commit 2224aff7ac
1 changed files with 17 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

22
2022/src/bin/day05.rs
View file

@ -18,14 +18,16 @@ struct Ship {
impl Ship {
fn parse(input: &str) -> Self {
let lines: Vec<&str> = input.trim().split('\n').rev().collect();
let lines: Vec<&str> = input.trim_end().split('\n').rev().collect();
let num_stacks = lines[0].split(" ").count();
let mut ship = Ship {
stacks: (0..=8).map(|_| Vec::new()).collect(),
stacks: (0..num_stacks).map(|_| Vec::new()).collect(),
};
for line in &lines[1..] {
for stk in 0..=8 {
for stk in 0..num_stacks {
let i = 4 * stk + 1;
let contents = match line.get(i..i + 1) {
None => continue,
@ -85,7 +87,13 @@ fn part1(input: &str) -> String {
let sections: Vec<&str> = input.split("\n\n").collect();
let mut ship = Ship::parse(sections[0]);
let moves: Vec<Move> = sections[1].trim().split('\n').map(Move::parse).collect();
let moves: Vec<Move> = sections[1]
.trim_end()
.split('\n')
.map(Move::parse)
.collect();
dbg!(&ship);
for m in moves {
ship.do_move_1(m);
@ -98,7 +106,11 @@ fn part2(input: &str) -> String {
let sections: Vec<&str> = input.split("\n\n").collect();
let mut ship = Ship::parse(sections[0]);
let moves: Vec<Move> = sections[1].trim().split('\n').map(Move::parse).collect();
let moves: Vec<Move> = sections[1]
.trim_end()
.split('\n')
.map(Move::parse)
.collect();
for m in moves {
ship.do_move_2(m);