use std::cmp::Reverse;

use aoc_runner_derive::{aoc, aoc_generator};
#[aoc_generator(day1)]
pub fn input_generator(input: &str) -> Vec<usize> {
    input.split("\n\n").map(|chunk| chunk.lines().map(|line| line.trim().parse::<usize>().unwrap()).sum()).collect()
}

#[aoc(day1, part1)]
pub fn solve_part1(input: &[usize]) -> usize {
    input.iter().max().copied().unwrap_or(0)
}

#[aoc(day1, part2)]
pub fn solve_part2(input: &[usize]) -> usize {
    let mut scores: Vec<usize> = input.to_vec();
    scores.sort_by_key(|v| Reverse(*v));
    scores.iter().take(3).sum()
}