blob: 15b21bd312dfbf0a0f0dc7130b72c94189bd122f (
plain) (
blame)
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
|
-- Author: Yuchen Pei (me@ypei.me)
import Data.Char (digitToInt)
solve1 :: [Char] -> Int
solve1 xs = let ys = digitToInt <$> xs in firstAndLast ys + solve' ys
solve' :: [Int] -> Int
solve' [] = 0
solve' [x] = 0
solve' (x:y:xs) = if x == y then x + solve' (y:xs) else solve' (y:xs)
firstAndLast :: [Int] -> Int
firstAndLast [] = 0
firstAndLast [x] = 0
firstAndLast xs = if head xs == last xs then head xs else 0
splitInput :: [Int] -> ([Int], [Int])
splitInput xs = splitAt (length xs `div` 2) xs
sumSame :: ([Int], [Int]) -> Int
sumSame ([], _) = 0
sumSame (_, []) = 0
sumSame (x:xs, y:ys) = if x == y then x * 2 + sumSame (xs, ys) else sumSame (xs, ys)
solve2 :: [Char] -> Int
solve2 xs = sumSame $ splitInput $ digitToInt <$> xs
|
