advent-of-code/2022/01/rust/main.rs

#![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")
        .map(|x| x.lines().map(|x| x.parse::<u32>().unwrap()).sum())
        .max()
        .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();
}