summaryrefslogtreecommitdiff
path: root/2023/day04/Main2.hs
diff options
context:
space:
mode:
Diffstat (limited to '2023/day04/Main2.hs')
-rw-r--r--2023/day04/Main2.hs26
1 files changed, 26 insertions, 0 deletions
diff --git a/2023/day04/Main2.hs b/2023/day04/Main2.hs
new file mode 100644
index 0000000..7b883e8
--- /dev/null
+++ b/2023/day04/Main2.hs
@@ -0,0 +1,26 @@
+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 :: [(Int, Int)] -> (Int, [(Int, Int)]) -> (Int, [(Int, Int)])
+calcPoints dict (num_cards, []) = (num_cards, [])
+calcPoints dict (num_cards, this) =
+ let new_wins = map (\(i, w) -> take w (drop i dict)) this
+ in calcPoints dict (num_cards + length new_wins, concat new_wins)
+
+
+
+main :: IO ()
+main = do
+ inputLines <- lines <$> getContents
+ let games = map (\x -> splitGame (getGame x)) inputLines
+ let wins = zip [1..] (map (\[w, cards] -> length (filter (\card -> elem card w) cards)) games) :: [(Int, Int)]
+ -- let points = map calcPoints wins
+
+ print (calcPoints wins (0, wins))
+ -- print (wins)
+