diff options
Diffstat (limited to '2024/1/p2.rs')
-rw-r--r-- | 2024/1/p2.rs | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/2024/1/p2.rs b/2024/1/p2.rs new file mode 100644 index 0000000..41eeea4 --- /dev/null +++ b/2024/1/p2.rs @@ -0,0 +1,25 @@ +use std::collections::HashMap; + +const INPUT_FILE: &str = "input"; + +fn main() { + let input = std::fs::read_to_string(INPUT_FILE).expect("Reading input file"); + let mut left: Vec<u32> = vec![]; + let mut right: HashMap<u32, u32> = HashMap::new(); + for line in input.lines() { + let mut fields = line.split_whitespace(); + let e = "Wrong file format"; + left.push(fields.next().expect(e).parse().expect(e)); + let r: u32 = fields.next().expect(e).parse().expect(e); + match right.get(&r) { + None => right.insert(r, 1), + Some(v) => right.insert(r, v+1), + }; + } + + let mut distance: u32 = 0; + for l in left { + distance += l * right.get(&l).unwrap_or(&0); + } + println!("Distance: {}", distance); +} |