feat(2024/Day25)

Author: goblivend

2024/Day25/Day25.hs

@@ -0,0 +1,43 @@
+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 Data.Tuple.Extra
+import Debug.Trace
+import System.Environment
+import Text.Regex.TDFA ((=~))
+-- TODO: Cleanup imports after day done
+
+type Input = ([[Int]], [[Int]])
+type Output = Int
+
+parseInput :: String -> Input
+parseInput input = (locks, keys)
+  where
+    inputs = splitOn "\n\n" input
+    filterOut c = map transpose $ filter (\e -> all (== c) $ head e) $ map lines inputs
+    findHeights c = map (map (((+) (-1)) . length . c))
+    locks = findHeights (takeWhile (== '#')) $ filterOut '#'
+    keys = findHeights (dropWhile (== '.')) $ filterOut '.'
+
+fit :: [Int] -> [Int] -> Bool
+fit k l = all (<= 5) . map (\(a, b) -> a+b) $ zip k l
+
+part1 :: Input -> Output
+part1 (locks, keys) = length [(key, lock) | key<-keys, lock<-locks, fit key lock]
+
+main :: IO ()
+main = do
+  args  <- getArgs
+  content <- readFile (last args)
+  let input = parseInput content
+
+  -- print input
+
+  print $ part1 input

2024/Day25/Makefile

@@ -0,0 +1,28 @@
+CONTAINER_NAME=haskell-aoc
+SRC=./Day25.hs
+TARGET=Day25
+all: $(TARGET)
+
+$(TARGET): $(SRC)
+	ghc -O3 $(SRC)
+
+profile: $(SRC)
+
+
+clean:
+	$(RM) ./Day25 ./Day25.o ./Day25.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