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

Day 10: Optimization

This commit is contained in:
shu 2020-12-11 00:28:13 +01:00
parent 9587504a13
commit 66fd0ab19e
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
2020/day10/day10.hs
View File

@ -6,12 +6,13 @@ main = do
input <- map read . lines <$> readFile "input" input <- map read . lines <$> readFile "input"
print $ (product . map length . group . sort . jolts) input print $ (product . map length . group . sort . jolts) input
print $ print $
(product . map (arrangements . sum) . filter (elem 1) . group . jolts) input (product .
map (last . trib [0, 0, 1] . sum) . filter (elem 1) . group . jolts)
input
arrangements :: Int -> Int jolts :: [Integer] -> [Integer]
arrangements =
(\x -> length . filter (== x)) <*>
(map sum . nub . map (filter (/= 0)) . flip replicateM [0 .. 3])
jolts :: [Int] -> [Int]
jolts = ap (zipWith subtract) tail . (\x -> 0 : x ++ [last x + 3]) . sort jolts = ap (zipWith subtract) tail . (\x -> 0 : x ++ [last x + 3]) . sort
trib :: [Integer] -> Integer -> [Integer]
trib x 0 = x
trib [x, y, z] n = trib [y, z, x + y + z] (n - 1)