2021/day10.hs

@@ -2,21 +2,18 @@ module Day10 where
 
 import Data.Maybe
 import Data.List (sort)
-import Control.Monad (foldM)
+closing :: Char -> Char
+closing '(' = ')'
+closing '[' = ']'
+closing '{' = '}'
+closing '<' = '>'
 
-closing :: Char -> Maybe Char
-closing '(' = Just ')'
-closing '[' = Just ']'
-closing '{' = Just '}'
-closing '<' = Just '>'
-closing _ = Nothing
-
-score :: Char -> Maybe Int
-score ')' = Just 3
-score ']' = Just 57
-score '}' = Just 1197
-score '>' = Just 25137
-score _ = Nothing
+score :: Char -> Int
+score ' ' = 0
+score ')' = 3
+score ']' = 57
+score '}' = 1197
+score '>' = 25137
 
 closingScore :: Char -> Int
 closingScore ')' = 1
@@ -31,14 +28,14 @@ calcScore xs = foldr (\char score -> score*5 + closingScore char) 0 (reverse xs)
 getInvalidChar :: [Char] -> String -> Char
 getInvalidChar _ [] = ' '
 getInvalidChar st (x:xs)
-    | x `elem` "([{<" = getInvalidChar (x:st) xs
-    | otherwise = if Just x == (listToMaybe st >>= closing) then getInvalidChar (tail st) xs else x
+    | x == '(' || x == '[' || x == '{' || x == '<' = getInvalidChar (x:st) xs
+    | otherwise = if Just x == fmap closing (listToMaybe st) then getInvalidChar (tail st) xs else x
 
 getCompletionString :: [Char] -> String -> String
-getCompletionString st [] = map (fromJust . closing) st
+getCompletionString st [] = map closing st
 getCompletionString st (x:xs)
-    | x `elem` "([{<" = getCompletionString (x:st) xs
-    | otherwise = if Just x == (listToMaybe st >>= closing) then getCompletionString (tail st) xs else ""
+    | x == '(' || x == '[' || x == '{' || x == '<' = getCompletionString (x:st) xs
+    | otherwise = if Just x == fmap closing (listToMaybe st) then getCompletionString (tail st) xs else ""
 
 
 main :: IO ()
@@ -46,7 +43,7 @@ main = do
     input <- lines <$> getContents
 
     putStr "part 1: "
-    let chars = mapMaybe (score . getInvalidChar []) input
+    let chars = map (score . getInvalidChar []) input
     print $ sum chars
 
     putStr "part 2: "