xiretza/advent-of-code
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

day16/haskell: prettify findSingletonChoice

Browse source
This commit is contained in:
Xiretza 2020-12-16 17:49:01 +01:00
parent 477f518531
commit 6dee5caf29
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
2020/day16/day16.hs
View file

@ -66,11 +66,12 @@ resolve = sequence . map fromResolved . iterateUntilNothing resolve' . map (Choi
fromResolved _ = Nothing fromResolved _ = Nothing
findSingletonChoice :: [Resolve a] -> Maybe (a, [Resolve a]) findSingletonChoice :: [Resolve a] -> Maybe (a, [Resolve a])
findSingletonChoice (c@(Choice s):xs) | S.size s == 1 = let [x] = S.elems s findSingletonChoice [] = Nothing
in Just (x, Resolved x:xs) findSingletonChoice (x:xs) =
| otherwise = (fmap.fmap) (c:) $ findSingletonChoice xs case x of
findSingletonChoice (x:xs) = (fmap.fmap) (x:) $ findSingletonChoice xs Choice s | S.size s == 1 -> let [r] = S.elems s
findSingletonChoice _ = Nothing in Just (r, Resolved r : xs)
_ -> fmap (\(r, rest) -> (r, x:rest)) $ findSingletonChoice xs
resolve' :: Ord a => [Resolve a] -> Maybe [Resolve a] resolve' :: Ord a => [Resolve a] -> Maybe [Resolve a]
resolve' xs = do resolve' xs = do