old stuff

2023/01/rust/main.rs

@@ -0,0 +1,103 @@
+#![feature(test)]
+
+extern crate test;
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use test::Bencher;
+
+    #[bench]
+    fn bench_a(b: &mut Bencher) {
+        b.iter(|| part_a(INPUT));
+    }
+
+    #[bench]
+    fn bench_b(b: &mut Bencher) {
+        b.iter(|| part_b(INPUT));
+    }
+}
+
+const INPUT: &str = include_str!("../input.txt");
+const TEST: &str = include_str!("../test.txt");
+
+fn part_a(input: &str) {
+    let result = input
+        .trim()
+        .split("\n")
+        .map(|line| {
+            let mut first: u16 = 0;
+            for char in line.chars() {
+                if first != 0 {
+                    break;
+                }
+                match char {
+                    '1' => first = 1,
+                    '2' => first = 2,
+                    '3' => first = 3,
+                    '4' => first = 4,
+                    '5' => first = 5,
+                    '6' => first = 6,
+                    '7' => first = 7,
+                    '8' => first = 8,
+                    '9' => first = 9,
+                    _ => continue,
+                }
+            }
+            let mut last: u16 = 0;
+            for char in line.chars().rev() {
+                if last != 0 {
+                    break;
+                }
+                match char {
+                    '1' => last = 1,
+                    '2' => last = 2,
+                    '3' => last = 3,
+                    '4' => last = 4,
+                    '5' => last = 5,
+                    '6' => last = 6,
+                    '7' => last = 7,
+                    '8' => last = 8,
+                    '9' => last = 9,
+                    _ => continue,
+                }
+            }
+            return first * 10 + last;
+        })
+        .sum::<u16>();
+    println!("{}", result);
+}
+
+fn part_b(input: &str) {
+    let result = input
+        .trim()
+        .split("\n")
+        .map(|line| {
+            let mut first: usize = 0;
+            let nums = [
+                "1", "2", "3", "4", "5", "6", "7", "8", "9", "one", "two", "three", "four", "five",
+                "six", "seven", "eight", "nine",
+            ];
+            for (i, s) in nums.iter().enumerate() {
+                if line.starts_with(s) {
+                    first = if i < 9 { i + 1 } else { i - 8 };
+                    break;
+                }
+                line.split_at(mid)
+            }
+            let mut last = 0 as usize;
+            return first * 10 + last;
+        })
+        .sum::<usize>();
+    println!("{}", result);
+}
+
+fn main() {
+    println!("Part A:");
+    part_a(TEST);
+    part_a(INPUT);
+
+    println!("\nPart B:");
+    part_b(TEST);
+    // part_b(INPUT);
+}