aoc
/
2022
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Day 5: Improve a lil more

This commit is contained in:
Nathan McCarty 2022-12-05 22:40:17 -05:00
parent 27d37e64b3
commit 7cb10255a3
Signed by: thatonelutenist
GPG Key ID: D70DA3DD4D1E9F96
1 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
05/Main2.idr
View File

@ -93,20 +93,35 @@ applyCommand multiple (Cmd count from to) stacks =
applyCommands : {n : Nat} -> Bool -> List (Command n) -> Stacks n -> Stacks n applyCommands : {n : Nat} -> Bool -> List (Command n) -> Stacks n -> Stacks n
applyCommands multiple commands stacks = foldl (\stack, command => applyCommand multiple command stack) stacks commands applyCommands multiple commands stacks = foldl (\stack, command => applyCommand multiple command stack) stacks commands
parseProblem : String -> Maybe (n ** (Stacks n, List (Command n)))
parseProblem content =
case last' . takeWhile(/= "") . lines $ content of
Just columnsLine =>
let size = foldl max 0 . catMaybes . map parsePositive . words $ columnsLine
stacks = parseStacks {n = size} . takeWhile (/= "") . lines $ content
commands = catMaybes . map (parseCommand {n = size}) . drop 1 . dropWhile (/= "") . lines $ content
in Just (size ** (stacks, commands))
Nothing => Nothing
doPart : {n : Nat} -> Bool -> Stacks n -> List (Command n) -> String
doPart multiple stacks commands = tops $ applyCommands multiple commands stacks
main : IO () main : IO ()
main = main =
do file <- readFile "input" do file <- readFile "input"
case file of case file of
Right content => Right content =>
let stacks = parseStacks {n = 9} . takeWhile (/= "") . lines $ content case parseProblem content of
commands = catMaybes . map (parseCommand {n = 9}) . drop 1 . dropWhile (/= "") . lines $ content Just (size ** (stacks, commands)) =>
part1 = tops $ applyCommands False commands stacks let part1 = doPart False stacks commands
part2 = tops $ applyCommands True commands stacks part2 = doPart True stacks commands
in do putStrLn "Input:" in do putStrLn "Input:"
printLn stacks printLn stacks
putStrLn "" putStr "\n\nPart 1: "
putStr "Part 1: "
putStrLn part1 putStrLn part1
putStr "Part 2: " putStr "Part 2: "
putStrLn part2 putStrLn part2
Nothing => putStrLn "Failed to parse problem"
Left err => printLn err Left err => printLn err