advent-of-code/2022/day01/src/main.rs

45 lines
1.2 KiB
Rust
Raw Normal View History

2022-12-15 23:51:46 +00:00
use std::env;
use std::fs;
use std::str::Lines;
2022-12-01 13:56:25 -08:00
2022-12-15 23:51:46 +00:00
fn get_calorie_totals(lines: Lines) -> Vec<u32> {
2022-12-01 13:56:25 -08:00
let mut elves: Vec<u32> = Vec::new();
let mut current_calorie_count: u32 = 0;
for line in lines {
if line.is_empty() {
2022-12-01 13:56:25 -08:00
elves.push(current_calorie_count);
current_calorie_count = 0;
continue;
}
let u32_value = u32::from_str_radix(&line, 10)
.expect(format!("Couldn't read u32 value from string: {}", line).as_str());
2022-12-01 13:56:25 -08:00
current_calorie_count += u32_value;
}
elves.sort_unstable_by(|a, b| b.cmp(a));
2022-12-15 23:51:46 +00:00
elves
2022-12-01 13:56:25 -08:00
}
2022-12-15 23:51:46 +00:00
fn main() {
let args: Vec<String> = env::args().collect();
let filename = args.get(1).expect("Missing filename argument");
let file_contents = fs::read_to_string(&filename).expect("Unable to read file");
let lines = file_contents.lines();
let elves = get_calorie_totals(lines);
2022-12-01 13:56:25 -08:00
2022-12-05 20:57:32 -08:00
println!(
"Part 1: Elf with highest calorie count in knapsack: {}",
elves[0]
);
2022-12-01 13:56:25 -08:00
2022-12-02 08:31:20 -08:00
let sum_of_top_three = &elves[0] + &elves[1] + &elves[2];
2022-12-05 20:57:32 -08:00
println!(
"Part 2: Elves with top 3 highest calorie counts in their knapsacks: {}, {}, {} = {}",
&elves[0], &elves[1], &elves[2], sum_of_top_three
);
2022-12-01 13:56:25 -08:00
}