day 15

alexjercan · Dec 15, 2023 · d87ad43 · d87ad43
1 parent a502540
commit d87ad43
Show file tree

Hide file tree

Showing 6 changed files with 111 additions and 3 deletions.
diff --git a/aoc.sh b/aoc.sh
@@ -69,3 +69,6 @@ stack exec day13 < input/day13.input
 
 echo -e "${IYellow}--- Day 14: Parabolic Reflector Dish ---${Color_Off}"
 stack exec day14 < input/day14.input
+
+echo -e "${IGreen}--- Day 15: Lens Library ---${Color_Off}"
+stack exec day15 < input/day15.input
diff --git a/aoc2023.cabal b/aoc2023.cabal
@@ -27,6 +27,7 @@ executable day01
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day02
@@ -41,6 +42,7 @@ executable day02
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day03
@@ -53,6 +55,7 @@ executable day03
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day04
@@ -67,6 +70,7 @@ executable day04
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day05
@@ -79,6 +83,7 @@ executable day05
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day06
@@ -91,6 +96,7 @@ executable day06
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day07
@@ -103,6 +109,7 @@ executable day07
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day08
@@ -117,6 +124,7 @@ executable day08
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day09
@@ -129,6 +137,7 @@ executable day09
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day10
@@ -141,6 +150,7 @@ executable day10
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day11
@@ -153,6 +163,7 @@ executable day11
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day12
@@ -165,6 +176,7 @@ executable day12
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day13
@@ -177,6 +189,7 @@ executable day13
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day14
@@ -189,10 +202,13 @@ executable day14
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day15
   main-is: Day15.hs
+  other-modules:
+      Util.Parser
   hs-source-dirs:
       src
   ghc-options: -Wall -Wcompat -Widentities -Wincomplete-record-updates -Wincomplete-uni-patterns -Wmissing-export-lists -Wmissing-home-modules -Wpartial-fields -Wredundant-constraints -main-is Day15
@@ -201,6 +217,7 @@ executable day15
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day16
@@ -213,6 +230,7 @@ executable day16
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day17
@@ -225,6 +243,7 @@ executable day17
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day18
@@ -237,6 +256,7 @@ executable day18
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day19
@@ -249,6 +269,7 @@ executable day19
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day20
@@ -261,6 +282,7 @@ executable day20
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day21
@@ -273,6 +295,7 @@ executable day21
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day22
@@ -285,6 +308,7 @@ executable day22
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day23
@@ -297,6 +321,7 @@ executable day23
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day24
@@ -309,6 +334,7 @@ executable day24
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 executable day25
@@ -321,6 +347,7 @@ executable day25
     , containers
     , parsec
     , split
+    , vector
   default-language: Haskell2010
 
 test-suite spec
@@ -367,6 +394,7 @@ test-suite spec
       TestDay12
       TestDay13
       TestDay14
+      TestDay15
       Paths_aoc2023
   autogen-modules:
       Paths_aoc2023
@@ -380,4 +408,5 @@ test-suite spec
     , hspec
     , parsec
     , split
+    , vector
   default-language: Haskell2010
diff --git a/package.yaml b/package.yaml
@@ -16,6 +16,7 @@ dependencies:
   - split
   - parsec
   - containers
+  - vector
 
 ghc-options:
 - -Wall
@@ -102,7 +103,7 @@ executables:
   day15:
     source-dirs:     src
     main:            Day15
-    other-modules:   []
+    other-modules:   [Util.Parser]
 
   day16:
     source-dirs:     src

diff --git a/src/Day15.hs b/src/Day15.hs
@@ -1,10 +1,61 @@
 module Day15 (main, part1, part2) where
 
+import Data.Char (ord)
+import qualified Data.Vector as V
+import qualified Data.Vector.Mutable as MV
+import qualified Text.Parsec as P
+import Util.Parser (Parser, parse)
+
+data Operation = Insert String Int | Delete String deriving (Show)
+
+csvP :: Parser [String]
+csvP = P.sepBy (P.many1 $ P.noneOf ",\n") (P.char ',') <* P.spaces <* P.eof
+
+insertP :: String -> Parser Operation
+insertP s = do
+    _ <- P.char '='
+    n <- P.many1 P.digit
+    return $ Insert s (read n)
+
+deleteP :: String -> Parser Operation
+deleteP s = do
+    _ <- P.char '-'
+    return $ Delete s
+
+operationP :: Parser Operation
+operationP = do
+    s <- P.many1 P.letter
+    insertP s P.<|> deleteP s
+
+operationsP :: Parser [Operation]
+operationsP = P.sepBy operationP (P.char ',') <* P.spaces <* P.eof
+
+hash :: String -> Int
+hash = foldl (\acc c -> (acc + c) * 17 `mod` 256) 0 . map ord
+
 part1 :: String -> String
-part1 = id
+part1 = show . sum . map hash . parse csvP
+
+insertPair :: (String, Int) -> [(String, Int)] -> [(String, Int)]
+insertPair (s, n) [] = [(s, n)]
+insertPair (s, n) ((s', n') : xs)
+    | s == s' = (s, n) : xs
+    | otherwise = (s', n') : insertPair (s, n) xs
+
+operation :: V.Vector [(String, Int)] -> Operation -> V.Vector [(String, Int)]
+operation v (Insert s n) = V.modify (\v' -> MV.modify v' (insertPair (s, n)) (hash s)) v
+operation v (Delete s) = V.modify (\v' -> MV.modify v' (filter (\(s', _) -> s /= s')) (hash s)) v
+
+simulate :: [Operation] -> V.Vector [(String, Int)]
+simulate = foldl operation (V.replicate 256 [])
+
+power :: V.Vector [(String, Int)] -> Int
+power = V.sum . V.imap (\i -> sum . zipWith (\j (_, x) -> (i + 1) * j * x) [1 ..])
+
+input = "rn=1,cm-,qp=3,cm=2,qp-,pc=4,ot=9,ab=5,pc-,pc=6,ot=7"
 
 part2 :: String -> String
-part2 = const ""
+part2 = show . power . simulate . parse operationsP
 
 solve :: String -> String
 solve input = "Part 1: " ++ part1 input ++ "\nPart 2: " ++ part2 input ++ "\n"

diff --git a/test/Main.hs b/test/Main.hs
@@ -14,6 +14,7 @@ import TestDay11
 import TestDay12
 import TestDay13
 import TestDay14
+import TestDay15
 
 main :: IO ()
 main = do
@@ -31,3 +32,4 @@ main = do
     TestDay12.test
     TestDay13.test
     TestDay14.test
+    TestDay15.test
diff --git a/test/TestDay15.hs b/test/TestDay15.hs
@@ -0,0 +1,22 @@
+module TestDay15 (test) where
+
+import Day15
+import Test.Hspec
+
+input :: String
+input = "rn=1,cm-,qp=3,cm=2,qp-,pc=4,ot=9,ab=5,pc-,pc=6,ot=7"
+
+test1 :: Expectation
+test1 = part1 input `shouldBe` "1320"
+
+test2 :: Expectation
+test2 = part2 input `shouldBe` "145"
+
+test :: IO ()
+test = hspec $ do
+    describe "day15" $ do
+        describe "part1" $ do
+            it "should work for the examples" test1
+
+        describe "part2" $ do
+            it "should work for the examples" test2