From 3dda0ec7cb4477259a1be52565fe27e0312b9d3b Mon Sep 17 00:00:00 2001 From: tzlil Date: Fri, 15 Dec 2023 19:34:05 +0200 Subject: wip 5 --- 4/1.hs | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 4/1.hs (limited to '4/1.hs') diff --git a/4/1.hs b/4/1.hs new file mode 100644 index 0000000..275a04c --- /dev/null +++ b/4/1.hs @@ -0,0 +1,23 @@ +import Prelude +import Text.Parsec.String (Parser) +import Text.Parsec +import System.Environment +import Control.Monad +import qualified Data.Set as S + +type Card = (Int, S.Set Int, S.Set Int) + +numbers = S.fromList <$> ((many1 space) *> sepEndBy (read <$> many1 digit) (many1 space)) + +card :: Parser Card +card = (,,) <$> (string "Card" *> spaces *> (read <$> many1 digit) <* char ':') <*> numbers <* (char '|') <*> numbers + +cards = many1 card + +solution :: [Card] -> Int +solution xs = sum $ map f xs where + f (id,picked,winning) = case (length $ S.intersection picked winning) of + 0 -> 0 + n -> 2 ^ (pred n) + +main = liftM2 (>>=) readFile (((print . solution . either (error.show) id) .) . parse cards) =<< head <$> getArgs \ No newline at end of file -- cgit 1.4.1