X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=advent05%2FMain.hs;h=d8bb80651619e20b6408db47ece5cfb9e999f897;hb=583c9cf8e3ec1bce0bf568b977251b5bc0045b81;hp=002b8f58e802504820a20a9eeb585570a958e8d7;hpb=f8648c0ce55739dcadf19bbcdd56d5e48cd4f2e6;p=advent-of-code-23.git diff --git a/advent05/Main.hs b/advent05/Main.hs index 002b8f5..d8bb806 100644 --- a/advent05/Main.hs +++ b/advent05/Main.hs @@ -39,7 +39,7 @@ lowestLocation almanac seeds = l (Iv l _) = head locations followRequirements :: Almanac -> Requirement -> Requirement -followRequirements _ req@(Requirement "location" vals) = req +followRequirements _ req@(Requirement "location" _) = req followRequirements almanac (Requirement name vals) = followRequirements almanac newReq where aMap = almanac ! name @@ -49,22 +49,24 @@ followRequirements almanac (Requirement name vals) = useRule :: Rule -> Interval -> ([Interval], [Interval], [Rule]) useRule (Rule (Iv rl rh) d) (Iv xl xh) = (newResults, newVals, newRules) where newResults = - filter legalInterval [ Iv (min xl rl) (min xh (rl - 1)) -- input below rule - , Iv ((max xl rl) + d) ((min xh rh) + d)] -- input within rule - newVals = filter legalInterval [Iv (max xl (rh + 1)) (max xh rh)] -- input above rule - newRules = filter legalRule [Rule (Iv (max (xh + 1) rl) (max xh rh)) d] -- rule above input + filter legalInterval + [ Iv xl (rl - 1) -- input below rule + , Iv ((max xl rl) + d) ((min xh rh) + d)] -- input within rule + newVals = filter legalInterval + [Iv (rh + 1) xh] -- input above rule + newRules = filter legalRule + [Rule (Iv (xh + 1) rh) d] -- rule above input useRules :: [Rule] -> [Interval] -> [Interval] useRules [] vals = vals useRules _ [] = [] -useRules (r@(Rule (Iv rl rh) _):rs) (v@(Iv xl xh):vs) - | rh < xl = useRules rs (v:vs) - | xh < rl = v : useRules (r:rs) vs +useRules (r@(Rule rv _):rs) (v:vs) + | rv `allBelow` v = useRules rs (v:vs) + | v `allBelow` rv = v : useRules (r:rs) vs | otherwise = newResults ++ (useRules (newRules ++ rs) (newVals ++ vs)) where (newResults, newVals, newRules) = useRule r v - legalInterval :: Interval -> Bool legalInterval (Iv l h) = l <= h @@ -85,7 +87,7 @@ expandRanges seeds = fmap expandRange ranges tidyIntervals :: [Interval] -> [Interval] -tidyIntervals ivs0 = tidyIntervalsS $ sort ivs0 +tidyIntervals = tidyIntervalsS . sort tidyIntervalsS :: [Interval] -> [Interval] tidyIntervalsS [] = []