Improve Day 4

Author: slgeay

README.md

@@ -29,15 +29,15 @@ Solutions for [Advent of Code](https://adventofcode.com/) in [Rust](https://www.
 | [Day 1](./src/bin/01.rs) | `87.7µs` | `115.4µs` |
 | [Day 2](./src/bin/02.rs) | `245.2µs` | `552.3µs` |
 | [Day 3](./src/bin/03.rs) | `403.3µs` | `403.8µs` |
-| [Day 4](./src/bin/04.rs) | `37.5ms` | `2.1ms` |
+| [Day 4](./src/bin/04.rs) | `7.0ms` | `777.1µs` |
 | [Day 5](./src/bin/05.rs) | `552.9µs` | `373.3µs` |
 | [Day 6](./src/bin/06.rs) | `66.1µs` | `54.5ms` |
 | [Day 7](./src/bin/07.rs) | `2.3ms` | `928.1ms` |
 | [Day 8](./src/bin/08.rs) | `39.3µs` | `85.2µs` |
 | [Day 9](./src/bin/09.rs) | `8.1ms` | `9.3ms` |
 | [Day 10](./src/bin/10.rs) | `1.6ms` | `1.4ms` |
 
-**Total: 1047.82ms**
+**Total: 1016.00ms**
 <!--- benchmarking table --->
 
 ---

src/bin/04.rs

@@ -16,22 +16,26 @@ pub fn part_one(i: &str) -> Option<u32> {
     let l: Vec<Vec<char>> = i.lines().map(|s| s.chars().collect()).collect();
     Some((0..l.len()).fold(0, |a, i| {
         a + (0..l[i].len()).fold(0, |b, j| {
-            b + A!(l[i][j..j + 4].iter())
-                + A!(l[i][j - 3..j + 1].iter().rev())
-                + A!(l[i..i + 4].iter().map(|l| l[j]))
-                + A!(l[i - 3..i + 1].iter().rev().map(|l| l[j]))
-                + A!(l[i..i + 4].iter().enumerate().map(|(k, l)| l[j + k]))
-                + A!(l[i..i + 4].iter().enumerate().map(|(k, l)| l[j - k]))
-                + A!(l[i - 3..i + 1].iter().rev().enumerate().map(|(k, l)| l[j + k]))
-                + A!(l[i - 3..i + 1].iter().rev().enumerate().map(|(k, l)| l[j - k]))
+            b + if l[i][j] != 'X' {
+                0
+            } else {
+                A!(l[i][j..j + 4].iter())
+                    + A!(l[i][j - 3..j + 1].iter().rev())
+                    + A!(l[i..i + 4].iter().map(|l| l[j]))
+                    + A!(l[i - 3..i + 1].iter().rev().map(|l| l[j]))
+                    + A!(l[i..i + 4].iter().enumerate().map(|(k, l)| l[j + k]))
+                    + A!(l[i..i + 4].iter().enumerate().map(|(k, l)| l[j - k]))
+                    + A!(l[i - 3..i + 1].iter().rev().enumerate().map(|(k, l)| l[j + k]))
+                    + A!(l[i - 3..i + 1].iter().rev().enumerate().map(|(k, l)| l[j - k]))
+            }
         })
     }))
 }
 
 #[rustfmt::skip]
 pub fn part_two(i: &str) -> Option<u32> {
     let l:Vec<Vec<char>>=i.lines().map(|s| s.chars().collect()).collect();
-    Some((1..l.len()-1).map(|i|{(1..l[i].len()-1).map(|j|{[-1,1].map(|d|{matches!((0..3).map(|k|{l[i+k-1][(j as i32+(d*(k as i32-1)))as usize]}).collect::<String>().as_str(),"MAS"|"SAM")}).iter().all(|&x|x)as u32}).sum::<u32>()}).sum())
+    Some((1..l.len()-1).map(|i|{(1..l[i].len()-1).map(|j|{if l[i][j]!='A'{0}else{[-1,1].map(|d|{matches!((0..3).map(|k|{l[i+k-1][(j as i32+(d*(k as i32-1)))as usize]}).collect::<String>().as_str(),"MAS"|"SAM")}).iter().all(|&x|x)as u32}}).sum::<u32>()}).sum())
 }
 
 #[cfg(test)]