update day6

old/day06/Cargo.lock

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

old/day06/Cargo.toml

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

old/day06/src/main.rs

@@ -1,56 +0,0 @@
-use std::fs;
-
-fn main() {
-    let input = fs::read_to_string("input.txt").unwrap();
-    let input: Vec<_> = input.split('\n') // Separate the Time and Distance lines
-        .map(|line| {
-            line[line.find(':').unwrap() + 1..] // Drop "Time:" and "Distance:"
-                .split_whitespace() // Split the numbers into their own elements
-                .flat_map(|s| s.chars()).collect::<String>() // Combine the strings into a single one
-                .parse::<i64>().expect("Couldn't parse number") // Parse numbers into i32
-        }).collect(); // Collect into Vec
-
-    let time = input[0];
-    let dist = input[1];
-    let mut valid = 0;
-
-    for remaining_time in 0..time {
-        if (time - remaining_time) * remaining_time > dist {
-            valid += 1;
-        }
-    }
-
-    println!("{}", valid);
-}
-
-
-
-// use std::fs;
-
-// fn main() {
-//     let input = fs::read_to_string("input.txt").unwrap();
-//     let input: Vec<_> = input.split('\n') // Separate the Time and Distance lines
-//         .map(|line| {
-//             line[line.find(':').unwrap() + 1..] // Drop "Time:" and "Distance:"
-//                 .split_whitespace() // Split the numbers into their own elements.
-//                 .map(|num| num.parse::<i32>().expect("Couldn't parse number")) // Parse numbers into i32
-//                 .collect::<Vec<_>>()
-//         }).collect(); // collect into Vec
-
-//     let mut valid_total = 1;
-
-//     for round in 0..input.first().unwrap().len() {
-//         let time = input[0][round];
-//         let dist = input[1][round];
-//         let mut valid = 0;
-
-//         for remaining_time in 0..time {
-//             if (time - remaining_time) * remaining_time > dist {
-//                 valid += 1;
-//             }
-//         }
-//         valid_total *= valid;
-//     }
-
-//     println!("{}", valid_total);
-// }

src/day6.rs

@@ -0,0 +1,80 @@
+use aoc_runner_derive::{aoc, aoc_generator};
+#[aoc_generator(day6, part1)]
+fn parse(input: &str) -> Vec<Vec<i32>> {
+    let input: Vec<Vec<i32>> = input.split('\n') // Separate the Time and Distance lines
+        .map(|line| {
+            line[line.find(':').unwrap() + 1..] // Drop "Time:" and "Distance:"
+                .split_whitespace() // Split the numbers into their own elements.
+                .map(|num| num.parse::<i32>().expect("Couldn't parse number")) // Parse numbers into i32
+                .collect::<Vec<_>>()
+        }).collect::<Vec<_>>(); // collect into Vec
+
+        input
+}
+
+#[aoc(day6, part1)]
+fn part1(input: &Vec<Vec<i32>>) -> i32 {
+    let mut valid_total = 1;
+
+    for round in 0..input.first().unwrap().len() {
+        let time = input[0][round];
+        let dist = input[1][round];
+        let mut valid = 0;
+
+        for remaining_time in 0..time {
+            if (time - remaining_time) * remaining_time > dist {
+                valid += 1;
+            }
+        }
+        valid_total *= valid;
+    }
+
+    valid_total
+}
+
+#[aoc_generator(day6, part2)]
+fn parse_part2(input: &str) -> Vec<i64> {
+    let input: Vec<i64> = input.split('\n') // Separate the Time and Distance lines
+        .map(|line| {
+            line[line.find(':').unwrap() + 1..] // Drop "Time:" and "Distance:"
+                .split_whitespace() // Split the numbers into their own elements
+                .flat_map(|s| s.chars()).collect::<String>() // Combine the strings into a single one
+                .parse::<i64>().expect("Couldn't parse number") // Parse numbers into i32
+        }).collect(); // Collect into Vec
+
+        input
+}
+
+#[aoc(day6, part2)]
+fn part2(input: &Vec<i64>) -> i32 {
+    let time = input[0];
+    let dist = input[1];
+    let mut valid = 0;
+
+    for remaining_time in 0..time {
+        if (time - remaining_time) * remaining_time > dist {
+            valid += 1;
+        }
+    }
+
+    valid
+}
+
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    const EX: &str = r"Time:      7  15   30
+Distance:  9  40  200";
+
+    #[test]
+    fn part1_example() {
+        assert_eq!(part1(&parse(EX)), 288);
+    }
+
+    #[test]
+    fn part2_example() {
+        assert_eq!(part2(&parse_part2(EX)), 71503);
+    }
+}

src/lib.rs

@@ -1,3 +1,4 @@
+mod day6;
 mod day5;
 mod day3;
 mod day14;