From 8d68c258949053f64102b40e5207bb840d312783 Mon Sep 17 00:00:00 2001 From: eriedaberrie Date: Mon, 11 Dec 2023 22:33:07 -0800 Subject: [PATCH] 2023 day 12: star 1 --- 2023/12/Main1.hs | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) create mode 100644 2023/12/Main1.hs diff --git a/2023/12/Main1.hs b/2023/12/Main1.hs new file mode 100644 index 0000000..7038959 --- /dev/null +++ b/2023/12/Main1.hs @@ -0,0 +1,24 @@ +import Data.List.Split (splitOn) +import Data.List (group) +import Control.Applicative (liftA2) + +matchesConsecutive :: [Int] -> String -> Bool +matchesConsecutive xs = (==) xs . map length . filter ((==) '#' . head) . group + +getChars :: Char -> [Char] +getChars '?' = ['.', '#'] +getChars c = [c] + +allSprings :: String -> [String] +allSprings = foldr (liftA2 (:) . getChars) [""] + +solve :: [Int] -> String -> Int +solve records = length . filter (matchesConsecutive records) . allSprings + +parseLine :: String -> ([Int], String) +parseLine l = (map read . splitOn "," $ y, x) + where + (x:y:_) = splitOn " " l + +main :: IO () +main = print . sum . map (uncurry solve . parseLine) . lines =<< readFile "data.txt"