aoc

advent of code
git clone git://source.orangerot.dev:/aoc.git
Log | Files | Refs

Main.hs (704B)

      1 import Data.List.Split
      2 import Data.List
      3 
      4 toMapFn :: [(Int, Int)] -> Int -> Int
      5 toMapFn map seed  
      6   | Just (d,s) <- find (\(drs, srs) -> srs == seed) map = d
      7   | otherwise = seed
      8 
      9 main :: IO ()
     10 main = do
     11   lines <- lines <$> getContents
     12 
     13   let sections = splitOn [""] lines
     14   let seeds = map read $ tail $ splitOn [' '] (sections!!0!!0) :: [Int]
     15 
     16   let maps = map (\(_:xs) -> (map (\l -> map (\i -> (read i) :: Int) (splitOn [' '] l)) xs)) (tail sections)
     17   let myMaps = map (\m -> concat (map (\[drs, srs, r] -> zip [drs..(drs+r)] [srs..(srs+r)]) m)) maps
     18   let myMapFns = map (toMapFn) myMaps
     19   let myFn = foldr (.) id (reverse myMapFns)
     20 
     21   print (minimum (map myFn seeds))
     22 
     23   -- mapM_ putStrLn lines
     24