43 lines
1.0 KiB
Rust
43 lines
1.0 KiB
Rust
|
use crate::prelude::*;
|
||
|
|
|
||
|
|
pub fn solution() -> Solution {
|
||
|
|
Solution::new(Day::new(15, 2020), part_1, part_2).with_expected(441, 10613991)
|
||
|
|
}
|
||
|
|
|
||
|
|
fn solve(input: &str, target: usize) -> isize {
|
||
|
|
let mut numbers = HashMap::new();
|
||
|
|
let mut previous_number = (0, (0, 0));
|
||
|
|
|
||
|
|
for (index, number) in input
|
||
|
|
.trim()
|
||
|
|
.split(',')
|
||
|
|
.map(str::parse::<isize>)
|
||
|
|
.map(Result::unwrap)
|
||
|
|
.enumerate()
|
||
|
|
{
|
||
|
|
numbers.insert(number, (index, index));
|
||
|
|
previous_number = (number, (index, index));
|
||
|
|
}
|
||
|
|
|
||
|
|
for index in numbers.len()..target {
|
||
|
|
let next_num = previous_number.1 .1 - previous_number.1 .0;
|
||
|
|
let num = numbers
|
||
|
|
.entry(next_num as isize)
|
||
|
|
.or_insert_with(|| (index, index));
|
||
|
|
num.0 = num.1;
|
||
|
|
num.1 = index;
|
||
|
|
|
||
|
|
previous_number = (next_num as isize, *num);
|
||
|
|
}
|
||
|
|
|
||
|
|
*numbers.iter().max_by(|a, b| a.1 .1.cmp(&b.1 .1)).unwrap().0
|
||
|
|
}
|
||
|
|
|
||
|
|
fn part_1(input: &str) -> Result<String> {
|
||
|
|
Ok(solve(input, 2020).to_string())
|
||
|
|
}
|
||
|
|
|
||
|
|
fn part_2(input: &str) -> Result<String> {
|
||
|
|
Ok(solve(input, 30000000).to_string())
|
||
|
|
}
|