10 part 1 done

This commit is contained in:
2024-12-10 08:14:57 +01:00
parent 73eec14746
commit 9b187ca14b
3 changed files with 74 additions and 4 deletions

View File

@ -2,14 +2,31 @@
module Main where
import Data.Functor
import Data.Char
import Data.Matrix hiding (trace)
import Data.List
parse = undefined
parse :: String -> Matrix Int
parse = fromLists . ((digitToInt <$>) <$>) . lines
solve1 = undefined
getAllDigit :: Int -> Matrix Int -> [(Int,Int)]
getAllDigit c s = zip [1..] (toLists s) >>= \(y,l) -> zip [1..] l >>= \(x,c') ->
if c == c' then pure (y,x) else mempty
getAbove :: (Int,Int) -> Matrix Int -> [(Int,Int)]
getAbove (y,x) s = let target = getElem y x s + 1 in
filter ((Just target ==) . flip (uncurry safeGet) s) [(y-1,x),(y+1,x),(y,x-1),(y,x+1)]
travel :: (Int,Int) -> Matrix Int -> [(Int,Int)]
travel p s = let curr = uncurry getElem p s in
if curr == 9 then [p] else getAbove p s >>= (`travel` s)
solve1 :: Matrix Int -> Int
solve1 s = length . concatMap (nub . (`travel` s)) $ getAllDigit 0 s
solve2 = undefined
main :: IO ()
main = readFile "inputs/10" <&> parse >>= \i ->
print (solve1 i) >>
print (solve2 i)
print (solve1 i)
-- >> print (solve2 i)