From b2f72b65b136d5e9360a5e756575be6738abbcf6 Mon Sep 17 00:00:00 2001 From: Amneesh Singh Date: Tue, 5 Dec 2023 16:13:40 +0530 Subject: [PATCH] day 4: minor optimisation Signed-off-by: Amneesh Singh --- day04.hs | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/day04.hs b/day04.hs index cc39b77..12f256c 100644 --- a/day04.hs +++ b/day04.hs @@ -43,7 +43,8 @@ main = do ) $ parsed where - -- TODO: dont bruteforce (its slow) consequence :: [Int] -> [Int] consequence [] = [] - consequence (x : xs) = 1 + sum (take x $ consequence xs) : consequence xs + consequence (x : xs) = + let xs' = consequence xs + in 1 + sum (take x xs') : xs'