day5, part1 done
This commit is contained in:
parent
a636bff3aa
commit
e7c4253f1c
3 changed files with 85 additions and 0 deletions
|
@ -8,3 +8,5 @@ edition = "2021"
|
|||
[dependencies]
|
||||
aoc-runner = "0.3.0"
|
||||
aoc-runner-derive = "0.3.0"
|
||||
itertools = "0.10.5"
|
||||
regex = "1.7.0"
|
||||
|
|
82
2022-aoc/src/d5.rs
Normal file
82
2022-aoc/src/d5.rs
Normal file
|
@ -0,0 +1,82 @@
|
|||
use aoc_runner_derive::{aoc as aoc_run, aoc_generator};
|
||||
use itertools::Itertools;
|
||||
|
||||
type Stax = Vec<Vec<char>>;
|
||||
type Inst = (u8, u8, u8);
|
||||
|
||||
enum Pstate {
|
||||
Setup,
|
||||
Instr,
|
||||
}
|
||||
|
||||
#[aoc_generator(day5)]
|
||||
fn parse_input_day1(input: &str) -> (Stax, Vec<Inst>) {
|
||||
let mut out = Vec::new();
|
||||
let mut inst = Vec::new();
|
||||
let mut state = Pstate::Setup;
|
||||
for _ in 0..9 {
|
||||
out.push(Vec::new());
|
||||
}
|
||||
|
||||
let re = regex::Regex::new(r"^move (\d+) from (\d+) to (\d+)$").unwrap();
|
||||
for line in input.lines() {
|
||||
match state {
|
||||
Pstate::Setup => {
|
||||
if line.trim().starts_with('1') {
|
||||
continue;
|
||||
}
|
||||
if line.trim().is_empty() {
|
||||
state = Pstate::Instr;
|
||||
}
|
||||
for (i, cell) in line.chars().chunks(4).into_iter().enumerate() {
|
||||
let cell: String = cell.collect();
|
||||
let cell = cell.trim();
|
||||
if !cell.is_empty() {
|
||||
let cell = cell.strip_prefix('[').unwrap().strip_suffix(']').unwrap();
|
||||
out[i].push(cell.chars().next().unwrap());
|
||||
}
|
||||
}
|
||||
}
|
||||
// parse the instructions
|
||||
Pstate::Instr => {
|
||||
let ins = re.captures(line).unwrap();
|
||||
inst.push((
|
||||
ins[1].parse().unwrap(),
|
||||
ins[2].parse::<u8>().unwrap() - 1,
|
||||
ins[3].parse::<u8>().unwrap() - 1,
|
||||
));
|
||||
}
|
||||
}
|
||||
}
|
||||
for stack in out.iter_mut() {
|
||||
stack.reverse();
|
||||
}
|
||||
(out, inst)
|
||||
}
|
||||
|
||||
#[aoc_run(day5, part1)]
|
||||
fn part1((stax, inst): &(Stax, Vec<Inst>)) -> String {
|
||||
let mut stax = stax.to_owned();
|
||||
for ins in inst.iter() {
|
||||
let (amt, from, to) = ins;
|
||||
let mut from = &mut stax[*from as usize];
|
||||
let mut tmp = Vec::with_capacity(*amt as usize);
|
||||
for _ in 0..*amt {
|
||||
let from = &mut from;
|
||||
let t = from.pop().unwrap();
|
||||
tmp.push(t);
|
||||
}
|
||||
stax[*to as usize].append(&mut tmp);
|
||||
}
|
||||
let mut s = String::with_capacity(9);
|
||||
for stack in stax {
|
||||
s.push(stack.last().copied().unwrap());
|
||||
}
|
||||
s
|
||||
}
|
||||
|
||||
#[aoc_run(day5, part2)]
|
||||
fn p2((stax, inst): &(Stax, Vec<Inst>)) -> u32 {
|
||||
//
|
||||
0
|
||||
}
|
|
@ -4,5 +4,6 @@ mod d1;
|
|||
mod d2;
|
||||
mod d3;
|
||||
mod d4;
|
||||
mod d5;
|
||||
|
||||
aoc_lib! { year = 2022 }
|
||||
|
|
Loading…
Reference in a new issue