eriedaberrie/advent-of-code
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

2023 day 8: touchup Haskell solutions

Browse source
This commit is contained in:
eriedaberrie 2023-12-11 20:14:45 -08:00
parent 504e5631fa
commit e3a9d35260
2 changed files with 15 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
2023/8/Main1.hs
View file

@ -16,11 +16,11 @@ countTimes directions m = countTimes' directions "AAA" 0
getDataMap :: [String] -> DataMap getDataMap :: [String] -> DataMap
getDataMap = HMS.fromList . map parseLine getDataMap = HMS.fromList . map parseLine
where where
parseLine l = (take 3 l, (take 3 $ drop 7 l, take 3 $ drop 12 l)) get3At x = take 3 . drop x
parseLine l = (get3At 0 l, (get3At 7 l, get3At 12 l))
main :: IO () main :: IO ()
main = do main = do
allLines <- lines <$> readFile "data.txt" allLines <- lines <$> readFile "data.txt"
let directions = cycle $ head allLines let directions = cycle . head $ allLines
dataMap = getDataMap $ drop 2 allLines print . countTimes directions . getDataMap . drop 2 $ allLines
print $ countTimes directions dataMap

15
2023/8/Main2.hs
View file

@ -15,17 +15,20 @@ countTimes directions m start = countTimes' directions start 0
| otherwise = countTimes' ds (getDirection d $ m HMS.! ss) . succ | otherwise = countTimes' ds (getDirection d $ m HMS.! ss) . succ
solve :: String -> DataMap -> Int solve :: String -> DataMap -> Int
solve directions m = foldl1' lcm $ map (countTimes directions m) $ filter ((== 'A') . head) $ HMS.keys m solve directions m = foldl1' lcm
. map (countTimes directions m)
. filter ((== 'A') . head)
. HMS.keys
$ m
getDataMap :: [String] -> DataMap getDataMap :: [String] -> DataMap
getDataMap = HMS.fromList . map parseLine getDataMap = HMS.fromList . map parseLine
where where
revTake3 = reverse . take 3 get3At x = reverse . take 3 . drop x
parseLine l = (revTake3 l, (revTake3 $ drop 7 l, revTake3 $ drop 12 l)) parseLine l = (get3At 0 l, (get3At 7 l, get3At 12 l))
main :: IO () main :: IO ()
main = do main = do
allLines <- lines <$> readFile "data.txt" allLines <- lines <$> readFile "data.txt"
let directions = cycle $ head allLines let directions = cycle . head $ allLines
dataMap = getDataMap $ drop 2 allLines print . solve directions . getDataMap . drop 2 $ allLines
print $ solve directions dataMap