benchmarks

2022/01/rust/main.rs

@@ -1,5 +1,24 @@
-fn main() {
-    // Part A
+#![feature(test)]
+
+extern crate test;
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use test::Bencher;
+
+    #[bench]
+    fn bench_a(b: &mut Bencher) {
+        b.iter(|| part_a());
+    }
+
+    #[bench]
+    fn bench_b(b: &mut Bencher) {
+        b.iter(|| part_b());
+    }
+}
+
+fn part_a() {
     let result: u32 = include_str!("../input.txt")
         .trim()
         .split("\n\n")
@@ -8,3 +27,19 @@ fn main() {
         .unwrap();
     println!("A: {}", result);
 }
+
+fn part_b() {
+    let mut result: Vec<u32> = include_str!("../input.txt")
+        .trim()
+        .split("\n\n")
+        .map(|x| x.lines().map(|x| x.parse::<u32>().unwrap()).sum::<u32>())
+        .collect();
+    result.sort();
+    let total = &result.as_slice()[result.len() - 3..result.len()];
+    println!("B: {:?}", total.iter().sum::<u32>());
+}
+
+fn main() {
+    part_a();
+    part_b();
+}