From de2df0f8c56f1db0f6e20d128c944168564b5183 Mon Sep 17 00:00:00 2001
From: Ishan Jain <contact@ishanjain.me>
Date: Thu, 1 Dec 2022 10:39:51 +0530
Subject: [PATCH] Added day 1 part 2

---
 src/main.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index ccf88c7..b0a0e12 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,7 +1,6 @@
-#![feature(test)]
-extern crate test;
+use std::cmp::Reverse;
 
-const INPUTS: [&'static str; 2] = [
+const INPUTS: [&str; 2] = [
     include_str!("../inputs/sample.txt"),
     include_str!("../inputs/input.txt"),
 ];
@@ -23,12 +22,13 @@ fn main() {
     for input in INPUTS.iter() {
         let output = parse(input);
 
-        let max = output
+        let mut sets: Vec<i64> = output
             .into_iter()
             .map(|c| c.into_iter().sum::<i64>())
-            .max()
-            .unwrap();
+            .collect();
 
-        println!("{}", max);
+        sets.select_nth_unstable_by_key(2, |c| Reverse(*c));
+
+        println!("{}", sets.into_iter().take(3).sum::<i64>());
     }
 }