AoC 2023HEADmain

1 files changed, 23 insertions, 0 deletions

diff --git a/2023/day05/MainOpt.hs b/2023/day05/MainOpt.hs
new file mode 100644
index 0000000..9b883e6
--- /dev/null
+++ b/2023/day05/MainOpt.hs
@@ -0,0 +1,23 @@
+import Data.List.Split
+import Data.List
+
+toMapFn :: [[Int]] -> Int -> Int
+toMapFn map seed  
+  | Just [d,s, r] <- find (\[drs, srs, r] -> (srs <= seed) && (seed <= srs + r)) map = d + (seed - s)
+  | 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 myMapFns = map (toMapFn) maps
+  let myFn = foldr (.) id (reverse myMapFns)
+
+  print (minimum (map myFn seeds))
+
+  -- mapM_ putStrLn lines
+