summaryrefslogtreecommitdiff
path: root/2023/rust/day11/src/lib.rs
diff options
context:
space:
mode:
authorSanto Cariotti <santo@dcariotti.me>2024-02-09 11:42:04 +0100
committerSanto Cariotti <santo@dcariotti.me>2024-02-09 11:42:04 +0100
commitdb2686c798378738020a0951a521358586e21aa0 (patch)
tree8d585ceb9bdc302ae10a2458cc343be40d60acb7 /2023/rust/day11/src/lib.rs
parent67dff579e91c13ef40ecb17d14497d145861c5e7 (diff)
Add day2 in OCaml
Diffstat (limited to '2023/rust/day11/src/lib.rs')
-rw-r--r--2023/rust/day11/src/lib.rs130
1 files changed, 130 insertions, 0 deletions
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);
+ // }
+}