advent-of-code/2023
advent-of-code/2023
1
0
Fork 0
Code Activity

day(17): use product of iterators

Browse source 
Signed-off-by: Matej Focko <me@mfocko.xyz>
This commit is contained in:
Matej Focko 2023-12-17 18:11:53 +01:00
parent 15fccb9f5e
commit 4c07bd4f2e
Signed by: mfocko
GPG key ID: 7C47D46246790496
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/bin/day17.rs
View file

@ -1,6 +1,7 @@
use std::collections::HashSet; use std::collections::HashSet;
use aoc_2023::*; use aoc_2023::*;
use itertools::iproduct;
type Output1 = u32; type Output1 = u32;
type Output2 = Output1; type Output2 = Output1;
@ -73,14 +74,15 @@ impl Day17 {
} }
seen.insert((s.position, s.direction)); seen.insert((s.position, s.direction));
for steps in min_steps..=max_steps { for (steps, d) in iproduct!(
for d in [left(&s.direction), right(&s.direction)] { min_steps..=max_steps,
if !in_range(&self.map, &(s.position + d * steps)) { [left(&s.direction), right(&s.direction)]
continue; ) {
} if !in_range(&self.map, &(s.position + d * steps)) {
continue;
queue.push(s.step(&self.map, d, steps));
} }
queue.push(s.step(&self.map, d, steps));
} }
} }