xiretza/advent-of-code
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

2022 day9/rust: abort early if knot doesn't move

Browse source
This commit is contained in:
Xiretza 2022-12-09 21:36:05 +01:00
parent e1862bdcbb
commit daad18c6af
1 changed files with 12 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
2022/day9/rust/src/main.rs
View file

@ -1,3 +1,4 @@
#![feature(iterator_try_reduce)]
#![warn(clippy::pedantic)]
use std::{
@ -7,15 +8,14 @@ use std::{
use aoc::vec2::{Direction, Vec2};
fn knot_movement(previous: Vec2, knot: Vec2) -> Vec2 {
fn knot_movement(previous: Vec2, knot: Vec2) -> Option<Vec2> {
let (dx, dy) = (previous - knot).into();
let movement = if dx.abs() <= 1 && dy.abs() <= 1 {
(0, 0)
if dx.abs() <= 1 && dy.abs() <= 1 {
None
} else {
(dx.signum(), dy.signum())
};
movement.into()
Some((dx.signum(), dy.signum()).into())
}
}
fn simulate_knots(num_knots: usize, instructions: &[(Direction, usize)]) -> usize {
@ -31,15 +31,12 @@ fn simulate_knots(num_knots: usize, instructions: &[(Direction, usize)]) -> usiz
for _i in 0..*distance {
knots[0] += Vec2::from(*direction);
let tail = knots
.iter_mut()
.reduce(|previous, knot| {
*knot += knot_movement(*previous, *knot);
knot
})
.unwrap();
tail_positions.insert(*tail);
if let Some(tail) = knots.iter_mut().try_reduce(|previous, knot| {
*knot += knot_movement(*previous, *knot)?;
Some(knot)
}) {
tail_positions.insert(tail.copied().unwrap());
}
}
}