From c6075f736e67e47b940abb9c1a0c54753d4b55b8 Mon Sep 17 00:00:00 2001 From: Orangerot Date: Fri, 17 May 2024 15:41:55 +0200 Subject: AoC 2023 --- 2023/day04/Main2.hs | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 2023/day04/Main2.hs (limited to '2023/day04/Main2.hs') 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) + -- cgit v1.2.3