From 779261f107669ef612eca264ab80a547529d8ead Mon Sep 17 00:00:00 2001 From: shu Date: Tue, 1 Dec 2020 12:03:49 +0100 Subject: [PATCH] Day 1: more golf --- 2020/day01/day01.hs | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/2020/day01/day01.hs b/2020/day01/day01.hs index c98a12c..33123c4 100644 --- a/2020/day01/day01.hs +++ b/2020/day01/day01.hs @@ -4,10 +4,10 @@ import Control.Monad main :: IO () main = do input <- map read . lines <$> readFile "input" - print $ day1 input 2 - print $ day1 input 3 + print $ day1 2 input + print $ day1 3 input -day1 :: [Int] -> Int -> Int -day1 input n = - snd . head . filter ((== 2020) . fst) $ - map (sum &&& product) $ replicateM n input +day1 :: Int -> [Int] -> Int +day1 = + fmap (snd . head . filter ((== 2020) . fst) . map (sum &&& product)) . + replicateM