fn solve_part_one(data: &str) -> u32 {
    data.split("\n\n")
        .collect::<Vec<&str>>()
        .iter()
        .map(|line| line.split("\n").flat_map(|x| x.parse::<u32>()).sum::<u32>())
        .max()
        .unwrap()
}

fn solve_part_two(data: &str) -> u32 {
    let mut vec = data
        .split("\n\n")
        .collect::<Vec<&str>>()
        .iter()
        .map(|line| line.split("\n").flat_map(|x| x.parse::<u32>()).sum::<u32>())
        .collect::<Vec<u32>>();
    vec.sort_by(|a, b| b.cmp(a));
    vec.iter().take(3).sum()
}

fn main() {
    let data = include_str!("../input/day1.prod");
    println!("{:?}", solve_part_one(data));
    println!("{:?}", solve_part_two(data));
}