day 4: rewrite

use Data.Text

Signed-off-by: Amneesh Singh <natto@weirdnatto.in>
This commit is contained in:
2022-12-09 17:26:45 +05:30
parent 42dcc56b13
commit 935a8300ca

35
day4.hs
View File

@@ -1,33 +1,28 @@
{-# LANGUAGE OverloadedStrings #-}
import Data.Char (isDigit) import Data.Char (isDigit)
import Text.ParserCombinators.ReadP import Data.Either (rights)
( ReadP, import Data.Maybe (fromJust)
char, import Data.Text (Text)
many1, import qualified Data.Text as T (lines, stripPrefix)
readP_to_S, import qualified Data.Text.Read as T (decimal)
satisfy, import Lib (readFile')
)
main :: IO () main :: IO ()
main = do main = do
input <- map (fst . last . readP_to_S parse) . lines <$> readFile "day4.in" input <- rights . map parse . T.lines <$> readFile' "day4.in"
putStr "Q1: " putStr "Q1: "
print $ q1 input print $ q1 input
putStr "Q2: " putStr "Q2: "
print $ q2 input print $ q2 input
parse :: ReadP ((Int, Int), (Int, Int)) parse :: Text -> Either String ((Int, Int), (Int, Int))
parse = do parse rest = do
a <- readInt (a, rest) <- T.decimal rest
char '-' (b, rest) <- T.decimal $ fromJust $ T.stripPrefix "-" rest
b <- readInt (c, rest) <- T.decimal $ fromJust $ T.stripPrefix "," rest
char ',' (d, rest) <- T.decimal $ fromJust $ T.stripPrefix "-" rest
c <- readInt
char '-'
d <- readInt
return ((a, b), (c, d)) return ((a, b), (c, d))
where
readInt :: ReadP Int
readInt = read <$> many1 (satisfy isDigit)
q1, q2 :: [((Int, Int), (Int, Int))] -> Int q1, q2 :: [((Int, Int), (Int, Int))] -> Int
q1 = length . filter (\((a, b), (c, d)) -> a >= c && b <= d || c >= a && d <= b) q1 = length . filter (\((a, b), (c, d)) -> a >= c && b <= d || c >= a && d <= b)