2024-01 rust p1,p2

1 files changed, 31 insertions, 0 deletions

diff --git a/2024/01/rust/src/main.rs b/2024/01/rust/src/main.rs
new file mode 100644
index 0000000..c1648cf
--- /dev/null
+++ b/2024/01/rust/src/main.rs
@@ -0,0 +1,31 @@
+use std::collections::HashMap;
+use std::io;
+
+fn main() -> io::Result<()> {
+    let (mut ls, mut rs): (Vec<u64>, Vec<u64>) = io::stdin()
+        .lines()
+        .map(|line| {
+            line.unwrap()
+                .split_whitespace()
+                .map(|s| s.parse().unwrap())
+                .collect::<Vec<u64>>()
+        })
+        .map(|xs| (xs[0], xs[1]))
+        .unzip();
+
+    ls.sort_unstable();
+    rs.sort_unstable();
+
+    let mut counts = HashMap::new();
+    rs.iter().for_each(|&x| {
+        *counts.entry(x).or_insert(0) += 1;
+    });
+
+    let silver: u64 = ls.iter().zip(rs).map(|(x, y)| x.abs_diff(y)).sum();
+    let gold: u64 = ls.iter().map(|x| x * counts.get(x).unwrap_or(&0)).sum();
+
+    println!("silver: {silver}");
+    println!("gold: {gold}");
+
+    return Ok(());
+}