diff options
Diffstat (limited to '2023/day05/Main.hs')
-rw-r--r-- | 2023/day05/Main.hs | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/2023/day05/Main.hs b/2023/day05/Main.hs new file mode 100644 index 0000000..e7c71e8 --- /dev/null +++ b/2023/day05/Main.hs @@ -0,0 +1,24 @@ +import Data.List.Split +import Data.List + +toMapFn :: [(Int, Int)] -> Int -> Int +toMapFn map seed + | Just (d,s) <- find (\(drs, srs) -> srs == seed) map = d + | otherwise = seed + +main :: IO () +main = do + lines <- lines <$> getContents + + let sections = splitOn [""] lines + let seeds = map read $ tail $ splitOn [' '] (sections!!0!!0) :: [Int] + + let maps = map (\(_:xs) -> (map (\l -> map (\i -> (read i) :: Int) (splitOn [' '] l)) xs)) (tail sections) + let myMaps = map (\m -> concat (map (\[drs, srs, r] -> zip [drs..(drs+r)] [srs..(srs+r)]) m)) maps + let myMapFns = map (toMapFn) myMaps + let myFn = foldr (.) id (reverse myMapFns) + + print (minimum (map myFn seeds)) + + -- mapM_ putStrLn lines + |