Add code for day 22 of 2024 (rust)

Author: loicdiridollou

2024/src/day21/bin/day21.rs

@@ -6,7 +6,7 @@ mod part1;
 mod part2;
 
 fn main() {
-    let _input = read_to_string("../input.txt").unwrap();
+    let _input = read_to_string("./day21/testinput.txt").unwrap();
 
     // start timer
     let start = Instant::now();

2024/src/day21/testinput.txt

@@ -0,0 +1,5 @@
+029A
+980A
+179A
+456A
+379A

2024/src/day22/bin/day22.rs

@@ -6,7 +6,7 @@ mod part1;
 mod part2;
 
 fn main() {
-    let _input = read_to_string("../input.txt").unwrap();
+    let _input = read_to_string("./day22/input.txt").unwrap();
 
     // start timer
     let start = Instant::now();

2024/src/day22/bin/part1.rs

@@ -1,7 +1,38 @@
 //! # Advent of Code - Day 22 - Part One
 
+use cached::proc_macro::cached;
+
+fn mix(new_value: i64, secret_number: i64) -> i64 {
+    return new_value ^ secret_number;
+}
+
+fn prune(new_value: i64) -> i64 {
+    return new_value.rem_euclid(16777216);
+}
+
+#[cached]
+pub fn generate_next(mut secret_num: i64) -> i64 {
+    secret_num = prune(mix(secret_num * 64, secret_num));
+    secret_num = prune(mix((secret_num as f64 / 32.0).floor() as i64, secret_num));
+    return prune(mix(secret_num * 2048, secret_num));
+}
+
 pub fn part1(_input: &str) -> usize {
-    return 0;
+    let secret_numbers = _input
+        .split("\n")
+        .filter(|c| !c.is_empty())
+        .map(|c| c.parse::<i64>().unwrap())
+        .collect::<Vec<i64>>();
+
+    let mut count = 0;
+    for mut secret_num in secret_numbers {
+        for _ in 0..2000 {
+            secret_num = generate_next(secret_num);
+        }
+        count += secret_num;
+    }
+
+    return count as usize;
 }
 
 #[cfg(test)]
@@ -10,6 +41,7 @@ mod day22 {
 
     #[test]
     fn test_part1() {
-        assert_eq!(part1(""), 0);
+        let _input = "1\n10\n100\n2024";
+        assert_eq!(part1(_input), 37327623);
     }
 }

2024/src/day22/bin/part2.rs

@@ -1,7 +1,39 @@
 //! # Advent of Code - Day 22 - Part Two
 
+use std::collections::{HashMap, HashSet};
+
+use crate::part1::generate_next;
+
 pub fn part2(_input: &str) -> usize {
-    return 0;
+    let secret_numbers = _input
+        .split("\n")
+        .filter(|c| !c.is_empty())
+        .map(|c| c.parse::<i64>().unwrap())
+        .collect::<Vec<i64>>();
+
+    let mut seq_total = HashMap::new();
+    for mut secret_num in secret_numbers {
+        let mut buyer = vec![secret_num % 10];
+        for _ in 0..2000 {
+            secret_num = generate_next(secret_num);
+            buyer.push(secret_num % 10);
+        }
+        let mut seen = HashSet::new();
+        for i in 0..(buyer.len() - 4) {
+            let diff = buyer[i..(i + 5)]
+                .windows(2)
+                .map(|window| window[1] - window[0])
+                .collect::<Vec<i64>>();
+            let diff_tup = (diff[0], diff[1], diff[2], diff[3]);
+            if seen.contains(&diff_tup) {
+                continue;
+            }
+            seen.insert(diff_tup);
+            *seq_total.entry(diff_tup).or_insert(0) += buyer[i + 4];
+        }
+    }
+
+    return *seq_total.iter().map(|(_, val)| val).max().unwrap() as usize;
 }
 
 #[cfg(test)]
@@ -10,6 +42,7 @@ mod day22 {
 
     #[test]
     fn test_part2() {
-        assert_eq!(part2(""), 0);
+        let _input = "1\n2\n3\n2024";
+        assert_eq!(part2(_input), 23);
     }
 }

2024/src/day22/testinput.txt

@@ -0,0 +1,4 @@
+1
+2
+3
+2024

README.md

@@ -37,9 +37,9 @@ chosen to learn Typescript. For 2024, we will do AoC in rust.
 | Day 17 | Chronospatial Computer | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day17/) | ⭐️⭐️ |
 | Day 18 |        RAM Run         | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day18/) | ⭐️⭐️ |
 | Day 19 |      Linen Layout      | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day19/) | ⭐️⭐️ |
-| Day 20 |                        | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day20/) |        |
-| Day 21 |                        | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day21/) |        |
-| Day 22 |                        | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day22/) |        |
+| Day 20 |     Race Condition     | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day20/) | ⭐️⭐️ |
+| Day 21 |    Keypad Conundrum    | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day21/) |        |
+| Day 22 |     Monkey Market      | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day22/) | ⭐️⭐️ |
 | Day 23 |                        | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day23/) |        |
 | Day 24 |                        | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day24/) |        |
 | Day 25 |                        | [Dir](https://gitlab.com/loicdiridollou/advent-of-code/-/blob/main/2024/src/day25/) |        |