speeeeed day4

Author: spaenleh

Sources/Day04.swift

@@ -1,71 +1,71 @@
 import Algorithms
 
 func searchInNeighbors(pos: Coord, _ grid: Grid) -> [Coord] {
-  pos.fullNeighbors.included(in: grid).filter { n in
-    grid[n] == "M"
-  }
+    pos.fullNeighbors.included(in: grid).filter { n in
+        grid[n] == "M"
+    }
 }
 
 func isMAS(pos: Coord, _ grid: Grid) -> Bool {
-  let candidates = pos.cornerNeighbors
-  if (candidates.allSatisfy { $0.isInside(grid: grid) }) {
-    let l = candidates.map { grid[$0] }
-    let letterSet = Set(l)
-    // same letter can not be on opposite side
-    return l[0] != l[2] && l[1] != l[3] && letterSet.count == 2 && letterSet == ["M", "S"]
-  }
-  return false
+    let candidates = pos.cornerNeighbors
+    if (candidates.allSatisfy { $0.isInside(grid: grid) }) {
+        let l = candidates.map { grid[$0] }
+        let letterSet = Set(l)
+        // same letter can not be on opposite side
+        return l[0] != l[2] && l[1] != l[3] && letterSet.count == 2 && letterSet == ["M", "S"]
+    }
+    return false
 }
 
 func isWhole(pos: Coord, inDir: Coord, grid: Grid) -> Bool {
-  let aPos = pos + inDir
-  let sPos = aPos + inDir
-  if aPos.isInside(grid: grid) && sPos.isInside(grid: grid) {
-    return grid[aPos] == "A" && grid[sPos] == "S"
-  }
-  return false
+    let aPos = pos + inDir
+    let sPos = aPos + inDir
+    if aPos.isInside(grid: grid) && sPos.isInside(grid: grid) {
+        return grid[aPos] == "A" && grid[sPos] == "S"
+    }
+    return false
 }
 
 struct Day04: AdventDay {
-  var data: String
-  var grid: Grid {
-    Grid(from: data)
-  }
-
-  func countAllWhere(letter search: Character, predicate: (Coord, Grid) -> Bool) -> Int {
-    grid.raw.enumerated().map { (y, line) in
-      line.enumerated().map { (x, letter) in
-        if letter == search {
-          return predicate(Coord(x: x, y: y), grid)
-        }
-        return false
-      }
-    }.flatMap { $0 }.count { $0 == true }
-  }
-
-  func part1() -> Int {
-    var count = 0
-    // get all positions of "X"
-    for (y, line) in grid.raw.enumerated() {
-      for (x, letter) in line.enumerated() {
-        if letter == "X" {
-          let currentPos = Coord(x: x, y: y)
-          let results = searchInNeighbors(pos: currentPos, grid)
-          if results.count > 0 {
-            for candidate in results {
-              let inDir = candidate - currentPos
-              if isWhole(pos: candidate, inDir: inDir, grid: grid) {
-                count += 1
-              }
+    var grid: Grid
+    init(data: String) {
+        grid = Grid(from: data)
+    }
+    
+    func countAllWhere(letter search: Character, predicate: (Coord, Grid) -> Bool) -> Int {
+        grid.raw.enumerated().map { (y, line) in
+            line.enumerated().map { (x, letter) in
+                if letter == search {
+                    return predicate(Coord(x: x, y: y), grid)
+                }
+                return false
+            }
+        }.flatMap { $0 }.count { $0 == true }
+    }
+    
+    func part1() -> Int {
+        var count = 0
+        // get all positions of "X"
+        for (y, line) in grid.raw.enumerated() {
+            for (x, letter) in line.enumerated() {
+                if letter == "X" {
+                    let currentPos = Coord(x: x, y: y)
+                    let results = searchInNeighbors(pos: currentPos, grid)
+                    if results.count > 0 {
+                        for candidate in results {
+                            let inDir = candidate - currentPos
+                            if isWhole(pos: candidate, inDir: inDir, grid: grid) {
+                                count += 1
+                            }
+                        }
+                    }
+                }
             }
-          }
         }
-      }
+        return count
+    }
+    
+    func part2() -> Int {
+        countAllWhere(letter: "A", predicate: isMAS)
     }
-    return count
-  }
-
-  func part2() -> Int {
-    countAllWhere(letter: "A", predicate: isMAS)
-  }
 }

Sources/Helpers/Grid.swift

@@ -25,6 +25,12 @@ extension Grid {
   init(from data: String) {
     self.raw = data.lines().map { Array($0) }
   }
+    
+    func toText() -> String {
+        String(self.raw.map {
+            String($0)
+        }.joined(by: "\n"))
+    }
 
   func includes(coord: Coord) -> Bool {
     coord.x >= 0 && coord.x <= self.width - 1 && coord.y >= 0 && coord.y <= self.height - 1