23 lines
456 B
Haskell
23 lines
456 B
Haskell
import Data.List (sortOn)
|
|
import Data.Ord (Down (Down))
|
|
|
|
main :: IO ()
|
|
main = do
|
|
input <- lines <$> readFile "day1.in"
|
|
putStr "Q1: "
|
|
print $ q1 input
|
|
putStr "Q2: "
|
|
print $ q2 input
|
|
|
|
q1 :: [String] -> Int
|
|
q1 = maximum . calories
|
|
|
|
q2 :: [String] -> Int
|
|
q2 = sum . take 3 . sortOn Down . calories
|
|
|
|
calories :: [String] -> [Int]
|
|
calories [] = []
|
|
calories input =
|
|
let (cur, rest) = break null input
|
|
in sum (map read cur) : calories (drop 1 rest)
|