Implement litGridCorners

lukasbicus · Oct 2, 2024 · 68a68cb · 68a68cb
1 parent be81b70
commit 68a68cb
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/scripts/aoc2015/day18/utils.test.ts b/scripts/aoc2015/day18/utils.test.ts
@@ -5,6 +5,7 @@ import {
   getNeighboursPositions,
   LightGrid,
   LightState,
+  litGridCorners,
   performStep,
 } from "./utils.ts";
 
@@ -78,3 +79,21 @@ describe("performStep", function () {
     assertEquals(performStep(initialGrid, size), expectedGrid);
   });
 });
+
+describe("litGridCorners", function () {
+  it("should lit grid corners", function () {
+    const { grid: initialGrid, size } = fillGrid(`.#.#.#
+...##.
+#....#
+..#...
+#.#..#
+####..`);
+    const { grid: expectedGrid } = fillGrid(`##.#.#
+...##.
+#....#
+..#...
+#.#..#
+####.#`);
+    assertEquals(litGridCorners(initialGrid, size), expectedGrid);
+  });
+});
diff --git a/scripts/aoc2015/day18/utils.ts b/scripts/aoc2015/day18/utils.ts
@@ -127,3 +127,15 @@ export function performStep(prevGrid: LightGrid, size: number) {
   // gets positions all neighbours of a light in a grid
   return grid;
 }
+
+export function litGridCorners(grid: LightGrid, size: number) {
+  const cornerPosition = size - 1;
+  grid.set(getGridKey({ x: 0, y: 0 }), LightState.TurnedOn);
+  grid.set(getGridKey({ x: 0, y: cornerPosition }), LightState.TurnedOn);
+  grid.set(getGridKey({ x: cornerPosition, y: 0 }), LightState.TurnedOn);
+  grid.set(
+    getGridKey({ x: cornerPosition, y: cornerPosition }),
+    LightState.TurnedOn,
+  );
+  return grid;
+}