-
Notifications
You must be signed in to change notification settings - Fork 15
/
day18.rs
108 lines (94 loc) · 3.66 KB
/
day18.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
//! # RAM Run
//!
//! We use a trick to speed things up. Instead of storing `#` and `.` in the grid, we store
//! the time when a block arrives. For example:
//!
//! ```none
//! ...#... ∞ ∞ ∞ 3 ∞ ∞ ∞
//! 5,4 ..#.... ∞ ∞ 4 ∞ ∞ ∞ ∞
//! 4,2 ....#.. ∞ ∞ ∞ ∞ 1 ∞ ∞
//! 4,5 => ....... => ∞ ∞ ∞ ∞ ∞ ∞ ∞
//! 3,0 .....#. ∞ ∞ ∞ ∞ ∞ 0 ∞
//! 2,1 ....#.. ∞ ∞ ∞ ∞ 2 ∞ ∞
//! ....... ∞ ∞ ∞ ∞ ∞ ∞ ∞
//! ```
//!
//! Now we can [BFS](https://en.wikipedia.org/wiki/Breadth-first_search) from any arbitrary
//! start time. Squares are safe if the grid time is greater than the start time.
//!
//! Part two uses an incremental flood fill, getting a little further each time and removing
//! blocking bytes in descending order of time until we reach the exit.
//!
//! * Start with `t = i32::MAX`
//! * Start flood fill from top-left origin.
//! * If we encounter a blocking byte with a time less than `t`
//! then push `(time, position)` onto a max heap keyed by time.
//! * If we exhaust the flood fill `VecDeque` then pop the heap's top item.
//! This is the oldest byte that we encountered blocking the way.
//! Set `t` to the byte's time and push position to the dequeue.
//! * Restart flood fill from new position until we reach the exit.
use crate::util::grid::*;
use crate::util::heap::*;
use crate::util::iter::*;
use crate::util::parse::*;
use crate::util::point::*;
use std::collections::VecDeque;
pub fn parse(input: &str) -> Grid<i32> {
let mut grid = Grid::new(71, 71, i32::MAX);
for (i, [x, y]) in input.iter_signed::<i32>().chunk::<2>().enumerate() {
grid[Point::new(x, y)] = i as i32;
}
grid
}
/// BFS from start to exit using a fixed time of 1024.
pub fn part1(grid: &Grid<i32>) -> u32 {
let mut grid = grid.clone();
let mut todo = VecDeque::new();
grid[ORIGIN] = 0;
todo.push_back((ORIGIN, 0));
while let Some((position, cost)) = todo.pop_front() {
if position == Point::new(70, 70) {
return cost;
}
for next in ORTHOGONAL.map(|o| position + o) {
if grid.contains(next) && grid[next] > 1024 {
grid[next] = 0;
todo.push_back((next, cost + 1));
}
}
}
unreachable!()
}
/// Incremental flood fill that removes one blocking byte at a time in descending order.
pub fn part2(grid: &Grid<i32>) -> String {
let mut time = i32::MAX;
let mut heap = MinHeap::new();
let mut grid = grid.clone();
let mut todo = VecDeque::new();
grid[ORIGIN] = 0;
todo.push_back(ORIGIN);
loop {
// Incremental flood fill that makes as much progress as possible.
while let Some(position) = todo.pop_front() {
if position == Point::new(70, 70) {
let index = grid.bytes.iter().position(|&b| b == time).unwrap() as i32;
return format!("{},{}", index % grid.width, index / grid.width);
}
for next in ORTHOGONAL.map(|o| position + o) {
if grid.contains(next) {
if time < grid[next] {
grid[next] = 0;
todo.push_back(next);
} else {
// Use negative value to convert min-heap to max-heap.
heap.push(-grid[next], next);
}
}
}
}
// Remove the latest blocking byte then try to make a little more progress in flood fill.
let (first, saved) = heap.pop().unwrap();
time = -first;
todo.push_back(saved);
}
}