rhiobet/advent-of-code-2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Slightly clean day 20

Browse Source
This commit is contained in:
RhiobeT Chan
2023-12-21 18:58:16 +01:00
parent 52b197b49e
commit b48a12c5cc
1 changed files with 25 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
day20/Part1.hs
View File

@@ -4,39 +4,42 @@ import Commons
import Data.Map (insert, foldr, (!), notMember) import Data.Map (insert, foldr, (!), notMember)
applyModule :: Modules -> Int -> Int -> String -> Bool -> String -> (String, [String], Bool, Modules, Int, Int) applyModule :: Modules -> String -> Bool -> String -> (String, [String], Bool, Modules, Int, Int)
applyModule modules low high source p n applyModule modules source p n
| notMember n modules = (n, [], True, modules, low, high) | notMember n modules = (n, [], True, modules, 0, 0)
| otherwise = | otherwise =
let m = modules ! n let m = modules ! n
in case m of in case m of
Broadcaster o -> (n, o, p, modules, if not p then low + length o else low, Broadcaster o -> (n, o, p, modules, if not p then length o else 0,
if p then high + length o else high) if p then length o else 0)
FlipFlop s o -> if not p then let newFF = FlipFlop {state = not s, outputs = o} FlipFlop s o -> if not p then let newFF = FlipFlop {state = not s, outputs = o}
in (n, o, not s, insert n newFF modules, in (n, o, not s, insert n newFF modules,
if s then low + length o else low, if s then length o else 0,
if not s then high + length o else high) if not s then length o else 0)
else (n, [], s, modules, low, high) else (n, [], s, modules, 0, 0)
Conjonction i o -> let newC = Conjonction {inputs = insert source p i, outputs = o} Conjonction i o -> let newC = Conjonction {inputs = insert source p i, outputs = o}
state = Data.Map.foldr (&&) True $ inputs newC state = Data.Map.foldr (&&) True $ inputs newC
in if state then (n, o, False, insert n newC modules, low + length o, high) in if state then (n, o, False, insert n newC modules, length o, 0)
else (n, o, True, insert n newC modules, low, high + length o) else (n, o, True, insert n newC modules, 0, length o)
applyModules :: Modules -> Int -> Int -> String -> Bool -> [String] -> ([(String, [String], Bool)], Modules, Int, Int) applyModules :: Modules -> String -> Bool -> [String] -> ([(String, [String], Bool)], Modules, Int, Int)
applyModules modules low high _ _ [] = ([], modules, low, high) applyModules modules _ _ [] = ([], modules, 0, 0)
applyModules modules low high source p (h: t) = applyModules modules source p (h: t) =
let (newS, outputs, newP, newModules, newLow, newHigh) = applyModule modules low high source p h let (newS, outputs, newP, newModules, low, high) = applyModule modules source p h
(result, newNewModules, newNewLow, newNewHigh) = applyModules newModules newLow newHigh source p t (result, newNewModules, newLow, newHigh) = applyModules newModules source p t
in ((newS, outputs, newP): result, newNewModules, newNewLow, newNewHigh) in ((newS, outputs, newP): result, newNewModules, low + newLow, high + newHigh)
applySteps :: Modules -> Int -> Int -> [(String, [String], Bool)] -> (Modules, Int, Int) applySteps :: Modules -> [(String, [String], Bool)] -> (Modules, Int, Int)
applySteps modules low high [] = (modules, low, high) applySteps modules [] = (modules, 0, 0)
applySteps modules low high ((source, names, pulse): t) = applySteps modules ((source, names, pulse): t) =
let (result, newModules, newLow, newHigh) = applyModules modules low high source pulse names let (result, newModules, low, high) = applyModules modules source pulse names
in applySteps newModules newLow newHigh (t ++ result) (newNewModules, newLow, newHigh) = applySteps newModules (t ++ result)
in (newNewModules, low + newLow, high + newHigh)
pressButton :: Modules -> (Modules, Int, Int) pressButton :: Modules -> (Modules, Int, Int)
pressButton modules = applySteps modules 1 0 [("button", ["broadcaster"], False)] pressButton modules =
let (newModules, low, high) = applySteps modules [("button", ["broadcaster"], False)]
in (newModules, low + 1, high)
pressButtonNTimes' :: Modules -> Int -> Int -> Int -> Int -> (Int, Int) pressButtonNTimes' :: Modules -> Int -> Int -> Int -> Int -> (Int, Int)
pressButtonNTimes' modules low high i n pressButtonNTimes' modules low high i n