rhiobet/advent-of-code-2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Day 10

Browse Source
This commit is contained in:
RhiobeT Chan
2023-12-10 16:21:01 +01:00
parent cb79c63bb2
commit b2eadb440e
5 changed files with 142 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
day10/Part2.hs Normal file
View File

@@ -0,0 +1,36 @@
module Part2 where
import Commons
import Data.Array ((!), bounds)
getAreaToStart :: (Int, Int) -> (Int, Int) -> Direction -> Direction -> Grid -> Int
getAreaToStart (y, x) (yMax, xMax) South sdir grid =
case grid ! (y, x) of
Pipe (North, South) -> xMax - x + getAreaToStart (y + 1, x) (yMax, xMax) South sdir grid
Pipe (North, East) -> getAreaToStart (y, x + 1) (yMax, xMax) East sdir grid
Pipe (North, West) -> xMax - x + getAreaToStart (y, x - 1) (yMax, xMax) West sdir grid
Start -> if sdir == South then xMax - x else 0
getAreaToStart (y, x) (yMax, xMax) West sdir grid =
case grid ! (y, x) of
Pipe (East, West) -> getAreaToStart (y, x - 1) (yMax, xMax) West sdir grid
Pipe (North, East) -> x - xMax - 1 + getAreaToStart (y - 1, x) (yMax, xMax) North sdir grid
Pipe (South, East) -> getAreaToStart (y + 1, x) (yMax, xMax) South sdir grid
Start -> 0
getAreaToStart (y, x) (yMax, xMax) North sdir grid =
case grid ! (y, x) of
Pipe (North, South) -> x - xMax - 1 + getAreaToStart (y - 1, x) (yMax, xMax) North sdir grid
Pipe (South, East) -> x - xMax - 1 + getAreaToStart (y, x + 1) (yMax, xMax) East sdir grid
Pipe (South, West) -> getAreaToStart (y, x - 1) (yMax, xMax) West sdir grid
Start -> if sdir == East then x - xMax - 1 else 0
getAreaToStart (y, x) (yMax, xMax) East sdir grid =
case grid ! (y, x) of
Pipe (East, West) -> getAreaToStart (y, x + 1) (yMax, xMax) East sdir grid
Pipe (North, West) -> getAreaToStart (y - 1, x) (yMax, xMax) North sdir grid
Pipe (South, West) -> xMax - x + getAreaToStart (y + 1, x) (yMax, xMax) South sdir grid
Start -> if sdir == South then xMax - x else 0
getCycleArea :: Grid -> Int
getCycleArea grid = let (dir, startCoords) = getStartDirection (getStartCoordinates grid) grid
((_, _), (yMax, xMax)) = bounds grid
in getAreaToStart startCoords (yMax, xMax) dir dir grid