diff options
author | Orangerot <purple@orangerot.dev> | 2024-05-17 15:41:55 +0200 |
---|---|---|
committer | Orangerot <purple@orangerot.dev> | 2024-05-17 15:41:55 +0200 |
commit | c6075f736e67e47b940abb9c1a0c54753d4b55b8 (patch) | |
tree | 69d827646e615e00f91ef54361cc1701e4f0eff7 /2023/day02/MySplit2.hs | |
parent | a1895fe157e06ee4d119576163ea76390b3d402c (diff) |
Diffstat (limited to '2023/day02/MySplit2.hs')
-rw-r--r-- | 2023/day02/MySplit2.hs | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/2023/day02/MySplit2.hs b/2023/day02/MySplit2.hs new file mode 100644 index 0000000..baee8ae --- /dev/null +++ b/2023/day02/MySplit2.hs @@ -0,0 +1,50 @@ +import Data.List.Split (splitOn) + +game :: String -> [String] +game = splitOn [';'] + +reveal :: String -> [String] +reveal x = concat ( map (splitOn [',']) (game x) ) + +colors :: String -> [(Int, String)] +colors x = map (\x -> color (splitOn [' '] x)) (reveal x) + +color :: [String] -> (Int, String) +color [_, i, s] = (read i, s) + +validColor :: (Int, String) -> Bool +validColor (i, s) = case s of + "red" -> i <= 12 + "green" -> i <= 13 + "blue" -> i <= 14 + _ -> False + +getColors :: String -> String +getColors s = tail (dropWhile (/= ':') s) + +isGameValid :: String -> Bool +isGameValid s = all validColor (colors (getColors s)) + +isValid :: [String] -> Bool +isValid s = all isGameValid s + +getMyColor :: String -> [(Int, String)] +getMyColor s = colors (getColors s) + +myFilter :: String -> [(Int, String)] -> Int +myFilter color s = maximum (map (\(i,_) -> i) (filter (\(_,c) -> c == color) s)) + +main :: IO () +main = do + inputLines <- lines <$> getContents + + -- let arePossible = filter isPossible lines + -- myColors <- isValid inputLines + -- print (sum (map (\(_,n) -> n) (filter (\(b,_) -> b) (zip (map isGameValid inputLines) [1..])))) + let myColors = map (\x -> getMyColor x) inputLines + let myResult = map (\c -> myFilter "red" c * myFilter "green" c * myFilter "blue" c) myColors + print (sum myResult) + + -- mapM_ putStrLn myColors + -- mapM_ putStrLn inputLines + |