feat(2024/Day10)

Author: goblivend

2024/Day10/Day10.hs

@@ -0,0 +1,54 @@
+module Main where
+
+import Data.Bits
+import Data.List
+import Data.List.Split
+import Data.List.Unique
+import Data.Matrix (Matrix, (!))
+import Data.Matrix qualified as Mat
+import Data.Set (Set)
+import Data.Set qualified as S
+import System.Environment
+-- TODO: Cleanup imports after day done
+
+type Input = (Matrix Int, [(Int, Int)])
+type Output = Int
+
+data Dir = UP | LEFT | RIGHT | DOWN deriving (Show, Eq)
+
+parseInput :: String -> Input
+parseInput s = (mat, starts)
+  where
+    mat = Mat.fromLists . map (map (read . (:[]))) $ lines s
+    starts = [(y, x) | y<-[1..Mat.nrows mat], x<-[1..Mat.ncols mat], mat ! (y,x) == 0]
+
+move :: (Int, Int) -> Dir -> (Int, Int)
+move (y, x) UP = (y-1, x)
+move (y, x) DOWN = (y+1, x)
+move (y, x) LEFT = (y, x-1)
+move (y, x) RIGHT = (y, x+1)
+
+isOut :: (Int, Int) -> (Int, Int) -> Bool
+isOut (width, height) (y, x) = x <= 0 || y <= 0 || x > width || y > height
+
+validEndTrails :: Matrix Int -> Int -> (Int, Int) -> [(Int, Int)]
+validEndTrails grid curr pos
+      | isOut (Mat.ncols grid, Mat.nrows grid) pos = []
+      | grid ! pos /= curr + 1 = []
+      | grid ! pos == 9 = [pos]
+      | otherwise = concat $ map (validEndTrails grid (grid ! pos) . move pos) [UP, DOWN, LEFT, RIGHT]
+
+part1 :: Input -> Output
+part1 (grid, pos) = sum $ map (length . sortUniq . validEndTrails grid (-1)) pos
+
+part2 :: Input -> Output
+part2 (grid, pos) = sum $ map (length . validEndTrails grid (-1)) pos
+
+main :: IO ()
+main = do
+  args  <- getArgs
+  content <- readFile (last args)
+  let input = parseInput content
+
+  print $ part1 input
+  print $ part2 input

2024/Day10/Makefile

@@ -0,0 +1,28 @@
+CONTAINER_NAME=haskell-aoc
+SRC=./Day10.hs
+TARGET=Day10
+all: $(TARGET)
+
+$(TARGET): $(SRC)
+	ghc -O3 $(SRC)
+
+profile: $(SRC)
+
+
+clean:
+	$(RM) ./Day10 ./Day10.o ./Day10.hi
+
+setup: $(SRC)
+	docker cp $(SRC) $(CONTAINER_NAME):/home/haskell/Main.hs
+	docker cp input.txt $(CONTAINER_NAME):/home/haskell/input.txt
+	docker cp shortinput.txt $(CONTAINER_NAME):/home/haskell/shortinput.txt
+
+profiling: setup
+	docker exec -it $(CONTAINER_NAME) cabal build --enable-profiling
+	docker exec -it $(CONTAINER_NAME) ./profile input.txt
+
+run: setup
+	docker exec -it $(CONTAINER_NAME) cabal build
+	docker exec -it $(CONTAINER_NAME) ./run input.txt
+
+.PHONY: all $(TARGET) clean setup profiling run

2024/Day10/input.txt

@@ -0,0 +1,40 @@
+6541001098012789610347890107654656710323
+7832102127643898701256521218323465891410
+8996543034556789650987434309012534892565
+3887689678965876501874345892105621763676
+4305678563456903416765676756898760654980
+5214107852107812321254382347872108901221
+6543236943056921010341291078963457654338
+7896545987045430010980012569454968983549
+3217830656189899121676101430356879892678
+2106921043210778234585232321267898761432
+3478854430345665056798743410456901050501
+4569763521012552143895654501345012347670
+3654012678903443212104309690432167898981
+2783656987654874908765218781201254012567
+1092347897893965889034765670387063013498
+1001298756102456776121874989496122110901
+2310891043201307655430923876565434325892
+3456780103011218967649810189410145456743
+2561078212320989858236702107320236787654
+1232569343423874749145893678741199899873
+0343454358514565632098704569632087684562
+0456789969609034501347612189323456893001
+1499876878798123101256543079012548762110
+2387905462687678871212344568187659450223
+3456012301056549960305650127691098321054
+3456732102345832154454781034540107650169
+2369847898738981023763692321121256743278
+1078456654567670119832103400012349894361
+0012387763456543208041076510123412765010
+7650196892565454589107889623296503854321
+8943256781074303673236908774387654983432
+8912965890985210984365219985345015676541
+7607834187866789875434308776236723498650
+6506543045679012766923105698109894567743
+5410432134988703457810014567056210754892
+0322345028767845893456723459847349889701
+1201276719454936712679801210738256776545
+2450989805103221604589752345629145480230
+2347823456012120113298943238710076591121
+1056910147893012320107654109656789432012

2024/Day10/shortinput.txt

@@ -0,0 +1,8 @@
+89010123
+78121874
+87430965
+96549874
+45678903
+32019012
+01329801
+10456732

2024/README.md

@@ -125,3 +125,9 @@ So I indeed took time to do some optimizations, now I run in 2s.
 The main reason for this time improvement is because I now play using a tuple (id, size) and not just the id, hence less items to loop through
 
 Tried more optimizations such as saving the first valid free spot but this was only slowing things down
+
+### Day 10:
+
+Today was so short that I didn't even take time to enjoy it...
+
+It was like asking 2+2...