From ca1e06cdb1d4ccf452b697e84f2abbdf927182d3 Mon Sep 17 00:00:00 2001
From: Anthony Oteri <anthony.oteri@gmail.com>
Date: Wed, 6 Dec 2023 11:01:28 -0500
Subject: [PATCH] Day 6 - Part 2

Signed-off-by: Anthony Oteri <anthony.oteri@gmail.com>
---
 day-06/src/main.rs | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/day-06/src/main.rs b/day-06/src/main.rs
index ac6d959..d4e5a29 100644
--- a/day-06/src/main.rs
+++ b/day-06/src/main.rs
@@ -55,9 +55,38 @@ fn part_1(input: &str) -> usize {
     results.iter().map(Vec::len).product()
 }
 
+#[instrument]
+fn part_2(input: &str) -> usize {
+    let input = part2_preformat(input);
+    let (_, races) = parse_times(&input).unwrap();
+
+    let results = races
+        .iter()
+        .map(|r| {
+            r.distance()
+                .into_iter()
+                .filter(|&d| d > r.record)
+                .collect::<Vec<u64>>()
+        })
+        .collect::<Vec<Vec<u64>>>();
+
+    results.iter().map(Vec::len).product()
+}
+
+fn part2_preformat(input: &str) -> String {
+    let input = String::from(input);
+    input
+        .lines()
+        .map(|l| l.replace(" ", ""))
+        .map(|l| l.replace(":", ": "))
+        .collect::<Vec<String>>()
+        .join("\n")
+}
+
 fn main() {
     let input = include_str!("../input.txt");
     println!("Part 1: {}", part_1(input));
+    println!("Part 2: {}", part_2(input));
 }
 
 #[cfg(test)]
@@ -66,8 +95,13 @@ mod tests {
 
     #[test_log::test]
     fn test_part_1() {
-        tracing::info!("Hello world");
         let input = include_str!("../test-input.txt");
         assert_eq!(part_1(input), 288);
     }
+
+    #[test_log::test]
+    fn test_part_2() {
+        let input = include_str!("../test-input.txt");
+        assert_eq!(part_2(input), 71503);
+    }
 }