advent04/Main.hs

   1 -- Writeup at https://work.njae.me.uk/2022/12/04/advent-of-code-2022-day-4/
   2
   3 import System.Environment
   4 import Data.Text (Text)
   5 import qualified Data.Text.IO as TIO
   6 import Data.Attoparsec.Text hiding (Result)
   7 -- import Control.Applicative
   8
   9 data Assignment = Assignment Int Int deriving (Show, Eq)
  10 type Pair = (Assignment, Assignment)
  11
  12 main :: IO ()
  13 main =
  14   do  dataFileName <- getDataFileName
  15       text <- TIO.readFile dataFileName
  16       let pairs = successfulParse text
  17       print $ part1 pairs
  18       print $ part2 pairs
  19
  20 getDataFileName :: IO String
  21 getDataFileName =
  22   do args <- getArgs
  23      progName <- getProgName
  24      let baseDataName =  if null args
  25                          then progName
  26                          else head args
  27      let dataFileName = "data/" ++ baseDataName ++ ".txt"
  28      return dataFileName
  29
  30 part1 :: [Pair] -> Int
  31 part1 = length . (filter hasContainment)
  32
  33 part2 :: [Pair] -> Int
  34 part2 = length . (filter overlaps)
  35
  36 hasContainment, disjoint, overlaps :: Pair -> Bool
  37 hasContainment (assignment1, assignment2) =
  38   (assignment1 `contains` assignment2) || (assignment2 `contains` assignment1)
  39
  40 disjoint (assignment1, assignment2) =
  41   (assignment1 `before` assignment2) || (assignment2 `before` assignment1)
  42
  43 overlaps pair = not $ disjoint pair
  44
  45 contains, before :: Assignment -> Assignment -> Bool
  46 contains (Assignment lower1 upper1) (Assignment lower2 upper2) =
  47   (lower1 <= lower2) && (upper1 >= upper2)
  48
  49 before (Assignment _lower1 upper1) (Assignment lower2 _upper2) = (upper1 < lower2)
  50
  51 -- Parse the input file
  52
  53 pairsP :: Parser [Pair]
  54 pairP :: Parser Pair
  55 assignmentP :: Parser Assignment
  56
  57 pairsP = pairP `sepBy` endOfLine
  58 pairP = (,) <$> assignmentP <* "," <*> assignmentP
  59
  60 assignmentP = Assignment <$> decimal <* "-" <*> decimal
  61
  62 successfulParse :: Text -> [Pair]
  63 successfulParse input =
  64   case parseOnly pairsP input of
  65     Left  _err -> [] -- TIO.putStr $ T.pack $ parseErrorPretty err
  66     Right pairs -> pairs