2022/rust/day01/src/main.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

use std::env;
use std::fs;

fn main() {
    let args = env::args().nth(1).unwrap();
    let input = fs::read_to_string(args).unwrap().trim().split("\n\n")
        .map(|x| x.trim().split("\n").map(|x| x.parse::<i32>().unwrap())
            .collect()).collect::<Vec<Vec<i32>>>();
    // println!("{:?}", input);

    let mut elves = input.iter().map(|x| x.iter().fold(0, |x,y| x + y))
        .collect::<Vec<i32>>();
    println!("{}", elves.iter().fold(0, |x,y| x.max(*y)));

    elves.sort_by(|x,y| y.cmp(x));
    println!("{}", elves[0] + elves[1] + elves[2]);
}