fuxino/AdventOfCode2024
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve code for Day 2, Puzzle 2

Browse Source
This commit is contained in:
daniele 2024-12-18 16:33:14 +01:00
parent fc4939c7f8
commit f9513efab4
Signed by: fuxino
GPG Key ID: 981A2B2A3BBF5514
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/Day02/Puzzle2.hs
View File

@ -1,6 +1,7 @@
module Day02.Puzzle2 (day02_2) where module Day02.Puzzle2 (day02_2) where
import Data.List (inits, sort, sortBy, tails) import Control.Monad
import Data.List (sort, sortBy)
import Data.Ord import Data.Ord
isSafe :: [Int] -> Bool isSafe :: [Int] -> Bool
@ -11,10 +12,9 @@ isSafe xs = (isAscending xs || isDescending xs) && maximum distances <= 3 && min
distances = map abs $ zipWith (-) xs (drop 1 xs) distances = map abs $ zipWith (-) xs (drop 1 xs)
removeLevel :: [Int] -> [[Int]] removeLevel :: [Int] -> [[Int]]
removeLevel xs = zipWith (++) ys zs removeLevel xs = filter (\x -> length x == l) $ filterM (const [True, False]) xs
where where
ys = map init $ drop 1 (inits xs) l = length xs - 1
zs = map (drop 1) $ init (tails xs)
day02_2 :: IO () day02_2 :: IO ()
day02_2 = do day02_2 = do