bert/adventofcode
1
0
Fork 0
mirror of https://github.com/bertptrs/adventofcode.git synced 2025-12-25 21:00:31 +01:00
Code Issues Projects Releases Wiki Activity

Fix day 15 part 1.

Browse Source
This commit is contained in:
Bert Peters
2018-12-15 21:12:58 +01:00
parent ae51e3c114
commit 10918a62ad
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
2018/src/day15.rs
View File

@@ -1,6 +1,7 @@
use std::cmp::Reverse;
use std::collections::BinaryHeap;
use std::collections::HashMap; use std::collections::HashMap;
use std::collections::HashSet; use std::collections::HashSet;
use std::collections::VecDeque;
use std::io::BufRead; use std::io::BufRead;
use std::io::BufReader; use std::io::BufReader;
use std::io::Read; use std::io::Read;
@@ -85,12 +86,12 @@ impl Day15 {
.filter(|x| x.is_alive()) .filter(|x| x.is_alive())
.map(|x| x.pos).collect(); .map(|x| x.pos).collect();
let mut todo = VecDeque::new(); let mut todo = BinaryHeap::new();
let mut prev: HashMap<Coordinate, Coordinate> = HashMap::new(); let mut prev: HashMap<Coordinate, Coordinate> = HashMap::new();
todo.push_back(initial); todo.push(Reverse((0, initial)));
while let Some((y, x)) = todo.pop_front() { while let Some(Reverse((d, (y, x)))) = todo.pop() {
let next = [ let next = [
(y - 1, x), (y - 1, x),
(y, x - 1), (y, x - 1),
@@ -102,10 +103,11 @@ impl Day15 {
if !prev.contains_key(pos) && !self.walls[pos.0][pos.1] { if !prev.contains_key(pos) && !self.walls[pos.0][pos.1] {
if enemy_positions.contains(pos) { if enemy_positions.contains(pos) {
return prev.remove(&(y, x)); return prev.remove(&(y, x));
} else if !all_positions.contains(pos) {; } else if !all_positions.contains(pos) {
;
let prev_step = *prev.get(&(y, x)).unwrap_or(pos); let prev_step = *prev.get(&(y, x)).unwrap_or(pos);
prev.insert(*pos, prev_step); prev.insert(*pos, prev_step);
todo.push_back(*pos); todo.push(Reverse((d + 1, *pos)));
} }
} }
} }