Loopless version

This commit is contained in:
Sebastian 2022-12-01 19:10:36 +01:00
parent 2a2043db7f
commit 9e0effe8e4
1 changed files with 9 additions and 17 deletions

View File

@ -7,23 +7,15 @@ fn main() -> Result<(), Box<dyn Error>> {
let file = File::open("inputs/day1.txt")?; let file = File::open("inputs/day1.txt")?;
let lines = io::BufReader::new(file).lines().map(|l| l.unwrap()); let lines = io::BufReader::new(file).lines().map(|l| l.unwrap());
let mut elves = Vec::<Vec<u32>>::new(); let mut calories_per_elf: Vec<u32> =
lines.map(|l| l.parse::<u32>()).fold(vec![0], |mut v, c| {
let mut current_elf = Vec::<u32>::new(); if let Ok(cal) = c {
for line in lines { *v.last_mut().unwrap() += cal;
if line != "" { } else {
let calories = line.parse().unwrap(); v.push(0);
current_elf.push(calories); };
} else { v
elves.push(current_elf); });
current_elf = Vec::<u32>::new();
}
}
let mut calories_per_elf: Vec<u32> = elves
.iter()
.map(|e| e.iter().fold(0, |a, b| a + b))
.collect();
calories_per_elf.sort(); calories_per_elf.sort();
calories_per_elf.reverse(); calories_per_elf.reverse();