Day9 Rust

day9/rust/.gitignore

@@ -0,0 +1 @@
+target

day9/rust/Cargo.lock

@@ -0,0 +1,7 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 3
+
+[[package]]
+name = "rust"
+version = "0.1.0"

day9/rust/Cargo.toml

@@ -0,0 +1,8 @@
+[package]
+name = "rust"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]

day9/rust/src/main.rs

@@ -0,0 +1,37 @@
+enum Direction {
+    Forward,
+    Backward,
+}
+
+fn main() {
+    let stdin = std::io::stdin();
+
+    let mut part1 = 0;
+    let mut part2 = 0;
+
+    for line in stdin.lines() {
+        let line = line.unwrap();
+
+        let history = line.split(" ").map(|e| e.parse().unwrap()).collect();
+
+        part1 += extrapolate(&history, Direction::Forward);
+        part2 += extrapolate(&history, Direction::Backward);
+    }
+
+    println!("{}", part1);
+    println!("{}", part2);
+}
+
+fn extrapolate(history: &Vec<isize>, direction: Direction) -> isize {
+    if history.iter().all(|e| *e == 0) {
+        return 0;
+    }
+
+    let new_history = history.windows(2).map(|pair| pair[1] - pair[0]).collect();
+
+    if let Direction::Forward = direction {
+        return history.last().unwrap() + extrapolate(&new_history, direction);
+    }
+
+    return history[0] - extrapolate(&new_history, direction);
+}