diff options
Diffstat (limited to '2023/day04/Main.hs')
-rw-r--r-- | 2023/day04/Main.hs | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/2023/day04/Main.hs b/2023/day04/Main.hs new file mode 100644 index 0000000..9bf100b --- /dev/null +++ b/2023/day04/Main.hs @@ -0,0 +1,20 @@ +import Data.List.Split (splitOn) + +getGame :: String -> String +getGame s = tail $ dropWhile (/= ':') s + +splitGame :: String -> [[String]] +splitGame s = map (\x -> filter (/= "") (splitOn [' '] x)) (splitOn ['|'] s) + +calcPoints :: [String] -> Int +calcPoints w = round $ 2^^(length w - 1) + +main :: IO () +main = do + inputLines <- lines <$> getContents + let games = map (\x -> splitGame (getGame x)) inputLines + let wins = map (\[w, cards] -> filter (\card -> elem card w) cards) games + let points = map calcPoints wins + + print (sum points) + |