Gattix/AoC
3
0
Fork 0
Code Issues 1 Pull Requests Releases Wiki Activity

Day 10: Make it less incomprehensible? 

I hope, at least
Lambdas are a bit easier to read I think
This commit is contained in:
shu 2020-12-10 09:38:15 +01:00
parent edc4c375ef
commit 13ecd8bad7
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
2020/day10/day10.hs
View File

@ -9,11 +9,9 @@ main = do
(product . map (arrangements . sum) . filter (elem 1) . group . jolts) input
arrangements :: Int -> Int
arrangements n =
length $
filter (== n) $ map sum $ nub $ map (filter (/= 0)) $ replicateM n [0 .. 3]
arrangements =
(\x -> length . filter (== x)) <*>
(map sum . nub . map (filter (/= 0)) . flip replicateM [0 .. 3])
jolts :: [Int] -> [Int]
jolts =
ap (zipWith subtract) tail .
ap (++) (flip (:) [] . (+ 3) . last) . (:) 0 . sort
jolts = ap (zipWith subtract) tail . (\x -> 0 : x ++ [last x + 3]) . sort