day1: update - avoid using unecessary helpers

Signed-off-by: Amneesh Singh <natto@weirdnatto.in>
This commit is contained in:
2022-12-05 20:05:55 +05:30
parent 664cc4f6f8
commit 7831d76641

18
day1.hs
View File

@@ -1,24 +1,22 @@
import Data.List (sortOn) import Data.List (sortOn)
import Data.Ord (Down (Down)) import Data.Ord (Down (Down))
import Lib (split)
main :: IO () main :: IO ()
main = do main = do
input <- readFile "day1.in" input <- lines <$> readFile "day1.in"
putStr "Q1: " putStr "Q1: "
print $ q1 input print $ q1 input
putStr "Q2: " putStr "Q2: "
print $ q2 input print $ q2 input
q1 :: String -> Int q1 :: [String] -> Int
q1 = maximum . calories q1 = maximum . calories
q2 :: String -> Int q2 :: [String] -> Int
q2 = sum . take 3 . sortOn Down . calories q2 = sum . take 3 . sortOn Down . calories
calories :: String -> [Int] calories :: [String] -> [Int]
calories = calories [] = []
map (sum . map read) calories input =
. filter (not . null) let (cur, rest) = break null input
. split "" in sum (map read cur) : calories (drop 1 rest)
. split '\n'