Baipyrus/AoC-23
1
0
Fork 0
mirror of https://github.com/Baipyrus/AoC-23.git synced 2024-12-25 19:41:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

implement p2 with direction toggle

Browse Source
This commit is contained in:
Baipyrus 2024-01-25 00:14:57 +01:00
parent f5ea37e0d8
commit 03a9838f4b
1 changed files with 46 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
src/day09/mod.rs
View File

@ -2,23 +2,37 @@ use crate::common::{convert_num, read_file, split_inputs, split_lines};
#[allow(dead_code)]
pub fn part_two() {
unimplemented!();
predict(false)
}
#[allow(dead_code)]
pub fn part_one() {
predict(true);
}
fn predict(forward: bool) {
let sequences = setup();
let predictions: Vec<i32> = sequences
.iter()
.map(|s| {
let current = s.to_vec();
let mut next = extra_diff(current);
expand(&mut next);
next.first()
.unwrap()
.last()
.unwrap()
.clone()
expand(&mut next, forward);
let history = next
.first()
.unwrap();
if forward {
history
.last()
.unwrap()
.clone()
} else {
history
.first()
.unwrap()
.clone()
}
})
.collect();
@ -29,7 +43,7 @@ pub fn part_one() {
println!("The sum of all prediction values is: '{sum}'.");
}
fn expand(strct: &mut Vec<Vec<i32>>) {
fn expand(strct: &mut Vec<Vec<i32>>, forward: bool) {
let last = strct
.last_mut()
.unwrap();
@ -48,17 +62,30 @@ fn expand(strct: &mut Vec<Vec<i32>>) {
.unwrap();
let next =
current
.last()
.unwrap()
.clone() +
last
.last()
.unwrap()
.clone();
current.push(next);
if forward {
current.push(
current
.last()
.unwrap()
.clone() +
last
.last()
.unwrap()
.clone()
);
} else {
current.insert(
0,
current
.first()
.unwrap()
.clone() -
last
.first()
.unwrap()
.clone()
);
}
}
}