From 40c5f7b10f4d4e819272fec95cc896f63a932cb3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ga=C3=ABl=20Depreeuw?= Date: Sun, 6 Dec 2020 23:20:41 +0100 Subject: [PATCH] Solve day 3, part 2 --- src/Day3.hs | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/Day3.hs b/src/Day3.hs index c72886b..04bbc2a 100644 --- a/src/Day3.hs +++ b/src/Day3.hs @@ -59,7 +59,13 @@ parseGrid s = day3 :: IO () day3 = do r <- readFile "./input/day3" + let grid = parseGrid r putStr "[Day 3-1] # trees: " - case slideDown (0, 0) (3, 1) (parseGrid r) (Just []) of + case slideDown (0, 0) (3, 1) grid (Just []) of Nothing -> putStrLn "ERROR" - Just locs -> print . countTrees $ locs \ No newline at end of file + Just locs -> print . countTrees $ locs + putStr "[Day 3-2] # trees: " + let mTrees = mapM (\x -> slideDown (0, 0) x grid (Just [])) [(1, 1), (3, 1), (5, 1), (7, 1), (1, 2)] + case mTrees of + Nothing -> putStrLn "ERROR" + Just trees -> print . product $ map countTrees trees