johannes/aoc_2018
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

day01-2 tuple

Browse Source
This commit is contained in:
Johannes
2018-12-01 17:54:30 +01:00
parent 2d64eefc09
commit 0494dee19d
1 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/tasks/day01.rs
View File

@@ -13,21 +13,18 @@ pub fn task1() {
pub fn task2() {
let contents = utils::read_file("input/day01.txt");
let mut seen: HashSet<i32> = HashSet::new();
seen.insert(0);
let start_state = (std::cell::RefCell::new(seen), std::cell::RefCell::new(0));
let seen: HashSet<i32> = vec![0].into_iter().collect();
let start_state = (seen, 0);
let final_state = contents
.lines()
.map(|line| line.parse::<i32>().unwrap())
.cycle()
.scan(&start_state, |(hash_set, current_freq), freq_change| {
*current_freq.borrow_mut() += freq_change;
if hash_set.borrow_mut().contains(&current_freq.borrow()) {
Some((current_freq.borrow(), true))
.scan(start_state, |state, freq_change| {
state.1 += freq_change;
if state.0.insert(state.1) {
Some((state.1, false))
} else {
let value: i32 = *current_freq.borrow_mut();
hash_set.borrow_mut().insert(value);
Some((current_freq.borrow(), false))
Some((state.1, true))
}
}).find(|state| state.1);