2017/01.hs

13 lines
469 B
Haskell
Raw Normal View History

2017-12-02 17:41:43 +00:00
circularSum :: [Int] -> Int
circularSum ns = snd $ foldr (\n (prev, sum) -> (n, n * fromEnum (n == prev) + sum)) (last ns, 0) ns
2017-12-02 17:41:43 +00:00
halfwaySum :: [Int] -> Int
halfwaySum ns =
let ms = drop (length ns `div` 2) ns ++ take (length ns `div` 2) ns
in foldr (\(n, m) sum -> n * fromEnum (n == m) + sum) 0 $ zip ns ms
2017-12-02 17:41:43 +00:00
main :: IO ()
main = do
nums <- fmap (map (read . pure)) $ readFile "01.txt"
2017-12-02 17:41:43 +00:00
print $ circularSum nums
print $ halfwaySum nums