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

Apparently monkey.id == m if you code it right

Browse source
This commit is contained in:
Jeremy Kaplan 2022-12-11 20:15:19 -08:00
commit db2aa419d8
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
2022/src/bin/day11.rs
View file

@ -7,6 +7,12 @@ const INPUT: &str = include_str!("../../input/day11.txt");
fn main() {
let monkeys: Vec<Monkey> = aoc::blocks(INPUT).map(Monkey::parse).collect();
// Assumptions:
for (i, m) in monkeys.iter().enumerate() {
assert!(i == m.id);
assert!(m.operation.is_some());
}
println!("Part 1: {}", monkey_business(monkeys.clone(), 20, 3));
println!("Part 2: {}", monkey_business(monkeys, 10_000, 1));
}
@ -17,6 +23,8 @@ fn monkey_business(mut monkeys: Vec<Monkey>, rounds: usize, confidence: i64) ->
for _ in 1..=rounds {
for m in 0..monkeys.len() {
// Don't forget to put the monkey back in the vec when we're done!
//
// This is a workaround for simultaneous mutable references to multiple vec items.
let mut monkey = std::mem::take(&mut monkeys[m]);
for mut item in monkey.items.drain(0..) {
@ -30,8 +38,7 @@ fn monkey_business(mut monkeys: Vec<Monkey>, rounds: usize, confidence: i64) ->
monkeys[dest].items.push_back(item);
}
let id = monkey.id;
monkeys[id] = monkey;
monkeys[m] = monkey;
}
}