summaryrefslogtreecommitdiff
path: root/2023/day02/MySplit.hs
blob: 2bb5f3ff81072907b21b3af389912cf42f191937 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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

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..]))))

  -- mapM_ putStrLn myColors
  -- mapM_ putStrLn inputLines