AoC 2023HEADmain

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)
+