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