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