summaryrefslogtreecommitdiff
path: root/2023/day05/MainOpt.hs
blob: 9b883e686a8ddd458815197c8eb80f83284b19bc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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