2023 day 8 part 1 in Haskell

2023/08/Day08.cabal

@@ -0,0 +1,19 @@
+cabal-version:   3.0
+name:            Day08
+version:         0.1.0.0
+license:         NONE
+build-type:      Simple
+extra-doc-files: CHANGELOG.md
+
+common warnings
+    ghc-options: -Wall
+
+executable Day08
+    import:           warnings
+    main-is:          Day08.hs
+    build-depends:    base ^>=4.17.2.1
+                    , attoparsec ^>=0.14.4
+                    , text ^>=2.1
+                    , containers ^>=0.6.7
+    hs-source-dirs:   .
+    default-language: Haskell2010

2023/08/Day08.hs

@@ -0,0 +1,47 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module Main where
+
+import Data.Attoparsec.Text as A
+import Data.Map
+import qualified Data.Text as T
+import qualified Data.Text.IO as TIO
+
+type Address = [Char]
+type Input = ([Char], Map Address (Address, Address))
+
+main :: IO ()
+main = do
+  input <- TIO.getContents >>= parseInput
+  putStrLn ("Part 1: " <> show (part1 input))
+  -- putStrLn ("Part 2: " <> show (part2 input))
+
+parseInput :: T.Text -> IO Input
+parseInput s = case parseOnly inputParser s of
+  Right i -> pure i
+  Left e -> error e
+
+start = "AAA"
+finish = "ZZZ"
+part1 (p, m) = walk (cycle p) m start 0
+
+walk _ _ "ZZZ" steps = steps
+walk (i:is) m src steps = walk is m next (1 + steps)
+  where
+    current = m ! src
+    next = case i of
+      'L' -> fst current
+      'R' -> snd current
+
+inputParser :: Parser Input
+inputParser = do
+  instructions <- takeTill isEndOfLine <* endOfLine <* endOfLine
+  tree <- many1 lineParser <* endOfInput
+  return (T.unpack instructions, fromList tree)
+
+lineParser :: Parser (Address, (Address, Address))
+lineParser = do
+  src <- A.take 3 <* string " = ("
+  l <- A.take 3 <* string ", "
+  r <- A.take 3 <* string ")" <* endOfLine
+  return (T.unpack src, (T.unpack l, T.unpack r))

2023/08/example

@@ -0,0 +1,9 @@
+RL
+
+AAA = (BBB, CCC)
+BBB = (DDD, EEE)
+CCC = (ZZZ, GGG)
+DDD = (DDD, DDD)
+EEE = (EEE, EEE)
+GGG = (GGG, GGG)
+ZZZ = (ZZZ, ZZZ)

2023/08/example2

@@ -0,0 +1,5 @@
+LLR
+
+AAA = (BBB, BBB)
+BBB = (AAA, ZZZ)
+ZZZ = (ZZZ, ZZZ)