diff --git a/Main.hs b/Main.hs index 9702924..e0ed13d 100644 --- a/Main.hs +++ b/Main.hs @@ -6,6 +6,7 @@ import Day1 import Day2 import Day3 import Day4 +import Day5 main :: IO () main = do @@ -18,4 +19,6 @@ main = do Day3.main putStrLn "Day 4" Day4.main + putStrLn "Day 5" + Day5.main diff --git a/flake.lock b/flake.lock index a5bd1eb..c278c7e 100644 --- a/flake.lock +++ b/flake.lock @@ -2,11 +2,11 @@ "nodes": { "nixpkgs": { "locked": { - "lastModified": 1703134684, - "narHash": "sha256-SQmng1EnBFLzS7WSRyPM9HgmZP2kLJcPAz+Ug/nug6o=", + "lastModified": 1703499205, + "narHash": "sha256-lF9rK5mSUfIZJgZxC3ge40tp1gmyyOXZ+lRY3P8bfbg=", "owner": "NixOS", "repo": "nixpkgs", - "rev": "d6863cbcbbb80e71cecfc03356db1cda38919523", + "rev": "e1fa12d4f6c6fe19ccb59cac54b5b3f25e160870", "type": "github" }, "original": { diff --git a/inputs/day5-test.input b/inputs/day5-test.input new file mode 100644 index 0000000..f756727 --- /dev/null +++ b/inputs/day5-test.input @@ -0,0 +1,33 @@ +seeds: 79 14 55 13 + +seed-to-soil map: +50 98 2 +52 50 48 + +soil-to-fertilizer map: +0 15 37 +37 52 2 +39 0 15 + +fertilizer-to-water map: +49 53 8 +0 11 42 +42 0 7 +57 7 4 + +water-to-light map: +88 18 7 +18 25 70 + +light-to-temperature map: +45 77 23 +81 45 19 +68 64 13 + +temperature-to-humidity map: +0 69 1 +1 0 69 + +humidity-to-location map: +60 56 37 +56 93 4 diff --git a/inputs/day5.input b/inputs/day5.input new file mode 100644 index 0000000..3d50c1c --- /dev/null +++ b/inputs/day5.input @@ -0,0 +1,252 @@ +seeds: 432986705 28073546 1364097901 88338513 2733524843 234912494 3151642679 224376393 485709676 344068331 1560394266 911616092 3819746175 87998136 892394515 435690182 4218056486 23868437 848725444 8940450 + +seed-to-soil map: +748585809 2125564114 88980459 +1317392128 775565564 217595062 +1218610825 676784261 98781303 +954230685 2235762425 141777617 +2920242079 4081180892 51765553 +2972007632 3159586797 16102841 +0 2377540042 17565155 +2834452876 3712797875 58062179 +2892515055 2917079842 6424918 +3327351062 3175689638 162608005 +673338549 647264576 29519685 +1197392973 2214544573 21217852 +738232750 116664417 10353059 +2988110473 2429807442 71556277 +17565155 334379348 277510712 +1700771639 228674051 105705297 +3059666750 4132946445 162020851 +1806476936 993160626 588628261 +1096008302 127289380 101384671 +622123656 1908836676 50942989 +3221687601 3338297643 28028532 +2408505336 3770860054 310320838 +4175210607 3039830108 119756689 +3326652416 3039131462 698646 +2898939973 2408505336 21302106 +673066645 127017476 271904 +3489959067 3382623558 330174317 +702858234 611890060 35374516 +4086270124 2562002619 88940483 +837566268 0 116664417 +1534987190 1959779665 165784449 +2718826174 2923504760 115626702 +3249716133 3366326175 16297383 +3820133384 2650943102 266136740 +3266013516 2501363719 60638900 +295075867 1581788887 327047789 + +soil-to-fertilizer map: +2018515973 2192795257 82329405 +3722326327 3015971185 249665840 +3046459770 3689390318 25519185 +3971992167 3265637025 40217941 +3071978955 3453653215 203407731 +0 443504340 17965088 +584437096 1722124969 470670288 +1055107384 744431503 164966659 +1489299099 461469428 282962075 +2321848831 2380372526 153650776 +2100845378 269225056 174279284 +3487660258 2648616968 234666069 +3275386686 3305854966 147798249 +1772261174 1172578553 246254799 +4012210108 2883283037 132688148 +3423184935 4138946628 64475323 +4144898256 2321848831 58523695 +538253726 1418833352 46183370 +1220074043 0 269225056 +17965088 909398162 263180391 +2590093273 3657060946 32329372 +281145479 1465016722 257108247 +2622422645 3714909503 424037125 +2475499607 2534023302 114593666 + +fertilizer-to-water map: +3731805434 353192162 37567806 +926873139 889685769 255250442 +3170336676 695153543 194532226 +679924479 451681440 193671776 +3009343704 3081959489 160992972 +1242360754 3579359343 278026518 +1861131448 2500688596 20068354 +4028837903 4006213119 266129393 +1182123581 3242952461 60237173 +3877550443 645353216 49800327 +2223776164 1371077033 341527178 +3364868902 2566566565 36440100 +1773121333 0 76664401 +264823995 2444756861 55931735 +3929841219 3857385861 27802851 +2166799431 1712604211 56976733 +873596255 1769580944 53276884 +645696746 3047731756 34227733 +3927350770 3955153621 2490449 +3769373240 177937131 108177203 +0 3314535348 264823995 +1942121274 3885188712 69964909 +1881199802 390759968 60921472 +1849785734 3303189634 11345714 +3401309002 2855740726 104355610 +2079164011 2960096336 87635420 +544424016 76664401 101272730 +2565303342 2520756950 45809615 +1520387272 2603006665 252734061 +2012086183 286114334 67077828 +2611112957 1822857828 398230747 +320755730 2221088575 223668286 +3505664612 1144936211 226140822 +4006213119 4272342512 22624784 + +water-to-light map: +62780592 544346201 30115959 +2740764032 1352944740 34082945 +377487729 807592920 35446631 +1316419610 1454554942 34907962 +986581913 756881718 50711202 +4167758628 3240047125 127208668 +818809239 1222506283 58684750 +3649838514 2036598113 6212644 +127663629 0 10715051 +3023280854 1435387310 19167632 +663070842 10715051 124076893 +2774846977 2422700597 37614763 +1812617371 2460315360 5121443 +1640337506 1864318248 172279865 +2986755724 1316419610 36525130 +2023334670 2467203928 540327060 +1159184084 248462172 14557802 +1037293115 152894449 95567723 +0 134791944 18102505 +18102505 712203631 44678087 +465375803 972369801 197695039 +2576394916 3007530988 65274194 +92896551 1281191033 34767078 +3656051158 4289142647 5824649 +412934360 1170064840 52441443 +3417303830 3873138614 84580361 +787147735 263019974 31661504 +1817738814 2042810757 205595856 +1285160111 843039551 30798000 +2563661730 3123136764 12733186 +138378680 305237152 239109049 +3648071389 2465436803 1767125 +1132860838 574462160 26323246 +888049663 873837551 98532250 +3626039273 3072805182 22032116 +3530183657 4193287031 95855616 +1404769450 3957718975 235568056 +3042448486 1489462904 374855344 +2641669110 1387027685 48359625 +877493989 294681478 10555674 +3501884191 3094837298 28299466 +1351327572 3186605247 53441878 +2690028735 3135869950 50735297 +2812461740 2248406613 174293984 +3661875807 3367255793 505882821 +1173741886 600785406 111418225 + +light-to-temperature map: +964570004 989608620 226759942 +2204148775 2545437438 20646474 +233260112 338444213 39032265 +958191857 332066066 6378147 +2318799855 914518254 75090366 +4247140372 3146297568 47826924 +2224795249 1216368562 94004606 +2871022952 1310373168 80313918 +1400254919 233260112 98805954 +445493256 487550555 149554087 +2576473348 3962746668 294549604 +3535295748 2775008885 371288683 +1499060873 377476478 110074077 +272292377 2215619580 173200879 +3347481948 1867953550 157067409 +4161267146 3794452372 85873226 +3504549357 2184873189 30746391 +1759636962 1780717197 87236353 +2951336870 2388820459 6114967 +1191329946 2566083912 208924973 +1884544339 3880325598 82421070 +595047343 3431307858 363144514 +2393890221 731935127 182583127 +4001414916 2025020959 159852230 +2957451837 1390687086 390030111 +1846873315 4257296272 37671024 +1966965409 3194124492 237183366 +1609134950 2394935426 150502012 +3906584431 637104642 94830485 + +temperature-to-humidity map: +1406768592 2335526312 13344484 +666958498 1862550129 472976183 +558853371 843618476 74696086 +1168798622 129171378 168640618 +1713291209 297811996 183431863 +1993628008 635748116 152317885 +2560263686 2849350774 11516524 +32266442 1212766321 287276323 +2571780210 3319898101 11192927 +375095240 995599149 183758131 +2661986290 2353962919 50829838 +3252020768 4280298713 14668583 +1337439240 1793220777 69329352 +3419718116 3502299739 574454544 +2353962919 2650392505 198958269 +633549457 1179357280 33409041 +2582973137 4076754283 50515665 +319542765 788066001 55552475 +1896723072 32266442 96904936 +1420113076 1500042644 293178133 +3006421020 2404792757 245599748 +2842554807 3331091028 163866213 +2633488802 2990605977 28497488 +2300450150 947178503 48420646 +3266689351 4127269948 153028765 +2145945893 481243859 154504257 +3994172660 3019103465 300794636 +1139934681 918314562 28863941 +2712816128 2860867298 129738679 +2552921188 3494957241 7342498 + +humidity-to-location map: +897459980 3171885613 268595078 +506368722 1864971513 13322696 +1166055058 2803961444 53745388 +2572095034 667166679 114420176 +687118932 1725187165 139784348 +2478398695 0 14138781 +3427672233 370325921 251085897 +3888215738 3612891343 82449665 +1674720770 1530101168 79955344 +3970665403 925512154 2812137 +519691418 2452425610 167427514 +3884704963 3168374838 3510775 +826903280 2381868910 70556700 +2399774019 349568762 20757159 +2972099388 3465151802 147739541 +1754676114 131614075 217954687 +2865104023 3440480691 24671111 +2206760431 932309368 77882935 +2284643366 1610056512 115130653 +2492537476 14138781 35151040 +2527688516 3695341008 44406518 +3119838929 781586855 143925299 +2732270071 2857706832 132833952 +1599442846 2728683520 75277924 +3263764228 3995626854 27783181 +0 2990540784 177834054 +2686515210 621411818 45754861 +2420531178 2670816003 57867517 +1219800446 1010192303 191374197 +3678758130 3789680021 205946833 +3973477540 3739747526 49932495 +1972630801 2014419033 234129630 +3291547409 1878294209 136124824 +2889775134 49289821 82324254 +1411174643 2619853124 50962879 +1466122599 2248548663 133320247 +177834054 1201566500 328534668 +1462137522 928324291 3985077 diff --git a/package.yaml b/package.yaml index 8d4cf8d..62e37cb 100644 --- a/package.yaml +++ b/package.yaml @@ -1,6 +1,6 @@ name: aoc23 -ghc-options: -Wall -threaded +ghc-options: -Wall -threaded #-prof -fprof-auto -rtsopts default-extensions: - OverloadedStrings @@ -11,6 +11,8 @@ dependencies: - bytestring - matrix - vector + - range-set-list + - containers executables: aoc23: @@ -25,4 +27,4 @@ library: - Day2 - Day3 - Day4 - + - Day5 diff --git a/src/Day5.hs b/src/Day5.hs new file mode 100644 index 0000000..44ebf51 --- /dev/null +++ b/src/Day5.hs @@ -0,0 +1,123 @@ +{-# LANGUAGE OverloadedStrings #-} + +{-# OPTIONS_GHC -Wno-unused-do-bind #-} + +module Day5 where + +import Debug.Trace + +import Data.Attoparsec.ByteString.Char8 +import qualified Data.ByteString as BS +import Data.RangeSet.List (RSet) +import qualified Data.RangeSet.List as RSet +import Data.Map.Strict (Map) +import qualified Data.Map.Strict as Map +import Data.Function ((&)) + +main :: IO () +main = do + putStrLn "Part 1 result:" + Right ra <- parseOnly rawAlmanach <$> BS.readFile "inputs/day5-test.input" + print $ part1 ra + putStrLn "Part 2 result:" + print $ part2 ra + +part1 :: RawAlmanach -> Int +part1 ra = (minimum . map (seedLocation ra) . rawSeeds) ra + +part2 :: RawAlmanach -> Int +part2 ra = minimum . map (seedLocation ra) . seedsRange $ ra + where + seedsRange al = concat $ fromRange (rawSeeds al) + fromRange (x:y:xs) = [x..x+y-1] : fromRange xs + fromRange [_] = error "Should not happen" + fromRange [] = [] + +-- Needs to be run ~2.3 billion times! +seedLocation :: RawAlmanach -> Int -> Int +seedLocation ra = mapSrcDst (combinedMaps ra) + +combinedMaps :: RawAlmanach -> AlmanachMap +combinedMaps ra = mergeMaps (rawHumidityToLoc ra) + $ mergeMaps (rawTempToHumidity ra) + $ mergeMaps (rawLightToTemp ra) + $ mergeMaps (rawWaterToLight ra) + $ mergeMaps (rawFertToWater ra) + $ mergeMaps (rawSoilToFert ra) (rawSeedToSoil ra) + +mapSrcDst :: AlmanachMap -> Int -> Int +mapSrcDst m src = case Map.lookup True $ Map.mapKeys (RSet.member src) m of + Just v -> src + v + Nothing -> src + +mergeMaps :: AlmanachMap -> AlmanachMap -> AlmanachMap +mergeMaps al1 = Map.foldlWithKey' f Map.empty + where + -- f acc k v = Map.union (mergeSet al1 k v) acc + f acc k v = Map.union (mergeSet al1 k v) acc + +mergeSet :: AlmanachMap -> RSet Int -> Int -> AlmanachMap +mergeSet m r mod' = Map.foldrWithKey' f Map.empty m + where + f k v acc = specialInsert right v acc + & specialInsert middle (v+mod') + & specialInsert left mod' + where + left = RSet.difference r k + middle = RSet.intersection k r + right = RSet.difference k r + specialInsert r' v' m' = case Map.lookup True $ Map.mapKeys ((== RSet.empty) . RSet.intersection r) m of + Just _ -> Map.insert r' v' m' + Nothing -> mergeSet m' r' v' + +data RawAlmanach = RawAlmanach + { rawSeeds :: [Int] + , rawSeedToSoil :: AlmanachMap + , rawSoilToFert :: AlmanachMap + , rawFertToWater :: AlmanachMap + , rawWaterToLight :: AlmanachMap + , rawLightToTemp :: AlmanachMap + , rawTempToHumidity :: AlmanachMap + , rawHumidityToLoc :: AlmanachMap + } + deriving Show + +type AlmanachMap = Map (RSet Int) Int + + +justSpace :: Parser BS.ByteString +justSpace = " " + +almanachMap :: Parser AlmanachMap +almanachMap = do + xs <- oneRange `sepBy` endOfLine + pure $ Map.fromList xs + where + oneRange = do + a:b:c:_ <- decimal `sepBy` justSpace + pure (RSet.fromRangeList [(b,b+c-1)], a-b) + +rawAlmanach :: Parser RawAlmanach +rawAlmanach = do + seeds <- "seeds: " *> decimal `sepBy` justSpace + endOfLine *> endOfLine + "seed-to-soil map:" *> endOfLine + sts <- almanachMap + endOfLine *> endOfLine + "soil-to-fertilizer map:" *> endOfLine + stf <- almanachMap + endOfLine *> endOfLine + "fertilizer-to-water map:" *> endOfLine + ftw <- almanachMap + endOfLine *> endOfLine + "water-to-light map:" *> endOfLine + wtl <- almanachMap + endOfLine *> endOfLine + "light-to-temperature map:" *> endOfLine + ltt <- almanachMap + endOfLine *> endOfLine + "temperature-to-humidity map:" *> endOfLine + tth <- almanachMap + endOfLine *> endOfLine + "humidity-to-location map:" *> endOfLine + RawAlmanach seeds sts stf ftw wtl ltt tth <$> almanachMap