2022 day 23 rust
This commit is contained in:
parent
96dfc59fac
commit
f6a90d3b73
|
@ -75,16 +75,13 @@ fn round_iter(data: &HashSet<Loc>) -> Box<dyn Iterator<Item=(bool, Vec<Loc>)>> {
|
|||
let mut elves = Vec::from_iter(data.iter().copied());
|
||||
let mut moves = VecDeque::from([move_n, move_s, move_w, move_e]);
|
||||
|
||||
Box::new(std::iter::repeat(())
|
||||
.map(move |_| {
|
||||
Box::new(std::iter::repeat(()).map(move |_| {
|
||||
let occupied = HashSet::from_iter(elves.iter().copied());
|
||||
let mut planned = HashMap::<Loc, Vec<usize>>::new();
|
||||
|
||||
for (index, &loc) in elves.iter().enumerate().filter(|&(_, loc)| has_neighbors(&occupied, *loc)) {
|
||||
if let Some(loc) = moves.iter().filter_map(|f| f(&occupied, loc)).next() {
|
||||
planned.entry(loc)
|
||||
.and_modify(|v| v.push(index))
|
||||
.or_insert(vec![index]);
|
||||
planned.entry(loc).and_modify(|v| v.push(index)).or_insert(vec![index]);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue