blob: 7dadbced8b00a5128cbf3c4722bc3a5dc1e20403 (
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
42
43
44
45
46
47
48
49
50
|
module Main where
isStartParen :: Char -> Bool
isStartParen '(' = True
isStartParen '[' = True
isStartParen '{' = True
isStartParen '<' = True
isStartParen _ = False
isEndParen :: Char -> Bool
isEndParen ')' = True
isEndParen ']' = True
isEndParen '}' = True
isEndParen '>' = True
isEndParen _ = False
flipParen :: Char -> Char
flipParen '(' = ')'
flipParen ')' = '('
flipParen '[' = ']'
flipParen ']' = '['
flipParen '{' = '}'
flipParen '}' = '{'
flipParen '<' = '>'
flipParen '>' = '<'
flipParen _ = undefined
getScore :: Char -> Int
getScore ')' = 3
getScore ']' = 57
getScore '}' = 1197
getScore '>' = 25137
consumeBlock :: Char -> String -> Either Char String
consumeBlock t "" = Right ""
consumeBlock t (c:cs) | c == flipParen t = Right cs
| isStartParen c = (consumeBlock c cs) >>= (consumeBlock t)
| otherwise = Left c
corruptionScore :: String -> Int
corruptionScore "" = 0
corruptionScore (c:cs) = case consumeBlock c cs of
Left c -> getScore c
Right s -> corruptionScore s
main :: IO ()
main = do
s <- readFile "./input"
let notCorrupted = foldr (+) 0 $ map corruptionScore (lines s)
print notCorrupted
|
