From db2686c798378738020a0951a521358586e21aa0 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Fri, 9 Feb 2024 11:42:04 +0100
Subject: Add day2 in OCaml

---
 2023/rust/day11/src/lib.rs | 130 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 130 insertions(+)
 create mode 100644 2023/rust/day11/src/lib.rs

(limited to '2023/rust/day11/src/lib.rs')

diff --git a/2023/rust/day11/src/lib.rs b/2023/rust/day11/src/lib.rs
new file mode 100644
index 0000000..66c95c4
--- /dev/null
+++ b/2023/rust/day11/src/lib.rs
@@ -0,0 +1,130 @@
+fn expand(m: &mut Vec<Vec<i16>>, ch: i16) {
+    let rows = m.len();
+    let mut cols = m[0].len();
+
+    let mut i = 0;
+    while i < rows {
+        if m[i].iter().filter(|&x| *x != 0).count() == 0 {
+            m.insert(i, vec![ch; cols]);
+            i += 1;
+        }
+        i += 1;
+    }
+    let rows = m.len();
+    i = 0;
+
+    while i < cols {
+        let mut c = 0;
+        for j in 0..rows {
+            if m[j][i] != 0 {
+                c += 1;
+            }
+        }
+
+        if c == 0 {
+            for j in 0..rows {
+                m[j].insert(i, ch);
+            }
+            cols += 1;
+            i += 1;
+        }
+        i += 1;
+    }
+}
+
+pub fn part1(input: &str) -> u32 {
+    let mut res: u32 = 0;
+
+    let mut m: Vec<Vec<i16>> = input
+        .trim_end()
+        .split('\n')
+        .map(|y| y.chars().map(|x| if x == '#' { 1 } else { 0 }).collect())
+        .collect();
+
+    expand(&mut m, 0);
+    let mut hs: Vec<(usize, usize)> = vec![];
+    let rows = m.len();
+    let cols = m[0].len();
+    for i in 0..rows {
+        for j in 0..cols {
+            if m[i][j] != 0 {
+                hs.push((i, j));
+            }
+        }
+    }
+
+    let n = hs.len();
+    for i in 0..n {
+        for j in i + 1..n {
+            res += ((hs[j].1 as i16 - hs[i].1 as i16).abs()
+                + (hs[j].0 as i16 - hs[i].0 as i16).abs()) as u32;
+        }
+    }
+
+    res
+}
+
+pub fn part2(input: &str) -> u32 {
+    let mut res: u32 = 0;
+
+    let mut m: Vec<Vec<i16>> = input
+        .trim_end()
+        .split('\n')
+        .map(|y| y.chars().map(|x| if x == '#' { 1 } else { 0 }).collect())
+        .collect();
+
+    expand(&mut m, 1);
+    let mut hs: Vec<(usize, usize)> = vec![];
+    let rows = m.len();
+    let cols = m[0].len();
+    for i in 0..rows {
+        for j in 0..cols {
+            if m[i][j] > 0 {
+                hs.push((i, j));
+            }
+        }
+    }
+
+    let n = hs.len();
+    for i in 0..n {
+        for j in i + 1..n {
+            res += ((hs[j].1 as i16 - hs[i].1 as i16).abs()
+                + (hs[j].0 as i16 - hs[i].0 as i16).abs()) as u32;
+        }
+    }
+
+    res
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example_part1() {
+        let input = include_str!("../example.txt");
+        let result = part1(input);
+        assert_eq!(result, 374);
+    }
+
+    #[test]
+    fn input_part1() {
+        let input = include_str!("../input.txt");
+        let result = part1(input);
+        assert_eq!(result, 9623138);
+    }
+
+    #[test]
+    fn example_part2() {
+        let input = include_str!("../example.txt");
+        let result = part2(input);
+        assert_eq!(result, 1030);
+    }
+    //
+    // #[test]
+    // fn input_part2() {
+    //     let input = include_str!("../input.txt");
+    //     let result = part2(input);
+    //     assert_eq!(result, 71585);
+    // }
+}
-- 
cgit v1.2.3-18-g5258