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