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
 |