wow this thing have more than one part

This commit is contained in:
Bizcochito 2022-12-01 11:09:43 +01:00
parent 4538a035c6
commit 2c055a1094
1 changed files with 41 additions and 4 deletions

View File

@ -6,10 +6,15 @@ fn main() {
println!("Hi this is the fist day of AOC2022, first we will read the file {}", FILE_PATH);
let contents = fs::read_to_string(FILE_PATH)
.expect("Should have been able to read the file");
println!("The elf with the biggest calories is: {}", get_max_elf(&contents));
println!("And we get the elfs from the file!");
let mut elfs = get_elfs(&contents);
println!("The elf with the biggest calories is carring: {}", get_max_elf(&elfs));
println!("The elf with the 3 biggest calories are carring: {}", get_max_3_elf(&mut elfs));
}
fn get_max_elf(input: &str) -> u32{
fn get_elfs(input: &str) -> Vec<u32> {
let binding = input.to_owned();
let input: Vec<&str> = binding.lines().collect();
let mut elfs = Vec::new();
@ -19,9 +24,22 @@ fn get_max_elf(input: &str) -> u32{
if line == "" {ptr+=1; elfs.push(0); continue;}
elfs[ptr] += u32::from_str(line).expect("all content must be numbers");
}
elfs
}
fn get_max_elf(elfs: &Vec<u32>) -> u32{
*elfs.iter().max().expect("there must be a maximum lol")
}
fn get_max_3_elf(elfs: &mut Vec<u32>) -> u32{
let mut out: u32 = 0;
elfs.sort();
for _ in 0..3 {
out += elfs.pop().expect("There must be at least 3 elfs");
}
out
}
#[cfg(test)]
mod test {
use super::*;
@ -41,7 +59,26 @@ mod test {
9000
10000"#;
assert_eq!(get_max_elf(input), 24000);
let tst = get_elfs(input);
assert_eq!(get_max_elf(&tst), 24000);
}
#[test]
fn it_works2() {
let input = r#"1000
2000
3000
4000
5000
6000
7000
8000
9000
10000"#;
let mut tst = get_elfs(input);
assert_eq!(get_max_3_elf(&mut tst), 45000);
}
}