From: Neil Smith Date: Mon, 11 Dec 2023 11:20:51 +0000 (+0000) Subject: Tidying X-Git-Url: https://git.njae.me.uk/?a=commitdiff_plain;ds=inline;h=1e96a9775f10096425b99308c3e5b9d21d2efbea;p=advent-of-code-23.git Tidying --- diff --git a/advent11/Main.hs b/advent11/Main.hs index cfdbe0e..2a94423 100644 --- a/advent11/Main.hs +++ b/advent11/Main.hs @@ -22,15 +22,13 @@ main = print $ part2 galaxies part1, part2 :: Galaxies -> Int -part1 galaxies = allDistances $ expandGalaxies galaxies 2 - -part2 galaxies = allDistances $ expandGalaxies galaxies (10^6) +part1 = allDistances . expandGalaxies 2 +part2 = allDistances . expandGalaxies (10^6) allDistances :: Galaxies -> Int allDistances gs = case S.minView gs of Nothing -> 0 - Just (g, gs') -> (S.foldl' ( - addDist g) 0 gs) + allDistances gs' + Just (g, gs') -> (S.foldl' (addDist g) 0 gs) + allDistances gs' where addDist g1 acc g2 = acc + distance g1 g2 distance :: Position -> Position -> Int @@ -38,8 +36,8 @@ distance g1 g2 = abs dr + abs dc where (V2 dr dc) = g1 ^-^ g2 -expandGalaxies :: Galaxies -> Int -> Galaxies -expandGalaxies galaxies scale = galaxies'' +expandGalaxies :: Int -> Galaxies -> Galaxies +expandGalaxies scale galaxies = galaxies'' where er = emptyRows galaxies galaxies' = expandRows galaxies er scale ec = emptyCols galaxies'