day 4: rewrite
use Data.Text Signed-off-by: Amneesh Singh <natto@weirdnatto.in>
This commit is contained in:
35
day4.hs
35
day4.hs
@@ -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)
|
||||||
|
Reference in New Issue
Block a user