Use HashMap Entry api in day 15

Slightly faster and a bit cleaner.
This commit is contained in:
Tyler Hallada 2020-12-15 10:52:36 -05:00
parent 52fafa5716
commit d311aec2e1

View File

@ -1,4 +1,5 @@
use anyhow::{anyhow, Result}; use anyhow::{anyhow, Result};
use std::collections::hash_map::Entry;
use std::collections::HashMap; use std::collections::HashMap;
use std::fs::File; use std::fs::File;
use std::io::prelude::*; use std::io::prelude::*;
@ -42,13 +43,16 @@ impl Game {
fn take_turn(&mut self) { fn take_turn(&mut self) {
self.turn += 1; self.turn += 1;
let new_num = if let Some(prev_turn) = self.prev_nums.get(&self.last_num) { self.last_num = match self.prev_nums.entry(self.last_num) {
Entry::Occupied(mut entry) => {
let prev_turn = entry.insert(self.turn - 1);
self.turn - prev_turn - 1 self.turn - prev_turn - 1
} else { }
Entry::Vacant(entry) => {
entry.insert(self.turn - 1);
0 0
}
}; };
self.prev_nums.insert(self.last_num, self.turn - 1);
self.last_num = new_num;
} }
} }