From bccd1d889ffca3981404fa4f3699d193524e120c Mon Sep 17 00:00:00 2001 From: Dominic Zimmer Date: Mon, 5 Dec 2022 13:25:10 +0100 Subject: [PATCH] Add day 5 --- 5.hs | 41 +++++ 5.in | 511 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 5.test | 10 ++ 3 files changed, 562 insertions(+) create mode 100644 5.hs create mode 100644 5.in create mode 100644 5.test diff --git a/5.hs b/5.hs new file mode 100644 index 0000000..c2b521d --- /dev/null +++ b/5.hs @@ -0,0 +1,41 @@ +{-# LANGUAGE ScopedTypeVariables #-} +import Data.Char +import Data.List.Split +import Data.List +import Data.Maybe + +main = do + ls <- readFile "5.in" + putStrLn . show . solve . parse $ ls where + solve (ss, ms) = map head . applyMovesA (map (map ((+)(-1))) ms) . map (filter (/=' ')) $ buildStacks ss where + --solveB (ss, ms) = map head . applyMovesB (map (map ((+)(-1))) ms) . map (filter (/=' ')) $ buildStacks ss + size = length $ ss !! 0 + buildStacks :: [[Char]] -> [[Char]] + buildStacks (x: xs) = zipWith (:) x (buildStacks xs) + buildStacks [] = replicate size [] + + + parse = parseStacksAndMoves . take 2 . map lines . splitOn "\n\n" where + parseStacksAndMoves [ss, ms] = (filter ((/='1') . (!!0) ) $ parseStacks ss, parseMoves ms) + parseStacks = map split3 where + split3 (a:b:c:[]) = b : [] + split3 (a:b:c:' ':xs) = b : (split3 xs) + split3 (_) = [] + parseMoves = map (map parseInt . getNums . words) where + getNums x = [x !! 1, x !! 3, x !! 5] + parseInt = read :: String -> Int + +applyMovesA moves stacks = foldl applyMove stacks moves where + applyMove stacks [count, from, to] = iterate (moveOne from to) stacks !! (count + 1) + moveOne from to stack = moveMany 0 from to stack + +applyMovesB moves stacks = foldl applyMove stacks moves where + applyMove stacks [count, from, to] = moveMany count from to stacks + +moveMany count from to stack = updateStack from to stack where + newItems = take (count +1) $ (stack !! from) + updateStack :: Int -> Int -> [[Char]] -> [[Char]] + updateStack from to (x:xs) | from == 0 = (iterate tail x !! (count + 1)) : (updateStack (from - 1) (to - 1) xs) + updateStack from to (x:xs) | to == 0= (newItems ++ x) : (updateStack (from - 1) (to - 1) xs) + updateStack from to (x:xs) = x : (updateStack (from - 1) (to - 1) xs) + updateStack from to [] = [] diff --git a/5.in b/5.in new file mode 100644 index 0000000..87fde8f --- /dev/null +++ b/5.in @@ -0,0 +1,511 @@ + [C] [N] [R] +[J] [T] [H] [P] [L] +[F] [S] [T] [B] [M] [D] +[C] [L] [J] [Z] [S] [L] [B] +[N] [Q] [G] [J] [J] [F] [F] [R] +[D] [V] [B] [L] [B] [Q] [D] [M] [T] +[B] [Z] [Z] [T] [V] [S] [V] [S] [D] +[W] [P] [P] [D] [G] [P] [B] [P] [V] + 1 2 3 4 5 6 7 8 9 + +move 4 from 9 to 6 +move 7 from 2 to 5 +move 3 from 5 to 2 +move 2 from 2 to 1 +move 2 from 8 to 4 +move 1 from 6 to 9 +move 1 from 9 to 4 +move 7 from 1 to 2 +move 5 from 2 to 3 +move 5 from 7 to 4 +move 5 from 6 to 3 +move 1 from 7 to 6 +move 2 from 6 to 9 +move 3 from 2 to 4 +move 4 from 5 to 6 +move 2 from 7 to 3 +move 2 from 9 to 3 +move 1 from 5 to 2 +move 11 from 4 to 3 +move 1 from 2 to 9 +move 1 from 9 to 3 +move 2 from 1 to 6 +move 5 from 8 to 5 +move 7 from 5 to 4 +move 2 from 5 to 6 +move 6 from 6 to 4 +move 17 from 3 to 4 +move 1 from 8 to 3 +move 11 from 4 to 7 +move 1 from 6 to 4 +move 3 from 4 to 2 +move 2 from 2 to 6 +move 8 from 3 to 1 +move 8 from 3 to 9 +move 3 from 9 to 6 +move 3 from 1 to 3 +move 11 from 7 to 5 +move 1 from 6 to 4 +move 4 from 9 to 6 +move 3 from 1 to 4 +move 1 from 2 to 3 +move 1 from 6 to 9 +move 24 from 4 to 9 +move 2 from 6 to 5 +move 1 from 1 to 2 +move 1 from 1 to 3 +move 12 from 9 to 6 +move 5 from 4 to 2 +move 4 from 2 to 3 +move 5 from 6 to 3 +move 13 from 6 to 7 +move 1 from 5 to 6 +move 9 from 5 to 3 +move 4 from 7 to 5 +move 1 from 6 to 1 +move 3 from 5 to 1 +move 14 from 9 to 4 +move 2 from 7 to 9 +move 13 from 4 to 9 +move 1 from 4 to 7 +move 4 from 7 to 9 +move 3 from 5 to 1 +move 8 from 3 to 9 +move 4 from 1 to 4 +move 8 from 3 to 7 +move 3 from 7 to 6 +move 4 from 4 to 2 +move 3 from 1 to 9 +move 6 from 2 to 6 +move 3 from 3 to 1 +move 7 from 9 to 7 +move 2 from 6 to 5 +move 1 from 5 to 3 +move 3 from 7 to 5 +move 5 from 7 to 4 +move 2 from 1 to 4 +move 5 from 5 to 9 +move 6 from 4 to 1 +move 6 from 7 to 8 +move 22 from 9 to 3 +move 7 from 1 to 8 +move 4 from 9 to 6 +move 1 from 4 to 5 +move 8 from 6 to 4 +move 7 from 8 to 1 +move 1 from 6 to 4 +move 1 from 9 to 4 +move 1 from 1 to 2 +move 1 from 2 to 5 +move 1 from 9 to 8 +move 11 from 3 to 7 +move 1 from 6 to 2 +move 2 from 1 to 5 +move 1 from 8 to 2 +move 1 from 7 to 8 +move 4 from 5 to 7 +move 1 from 6 to 9 +move 6 from 3 to 1 +move 6 from 3 to 1 +move 15 from 7 to 5 +move 1 from 3 to 1 +move 1 from 3 to 6 +move 1 from 6 to 8 +move 14 from 5 to 1 +move 16 from 1 to 3 +move 2 from 8 to 9 +move 1 from 7 to 4 +move 3 from 9 to 8 +move 3 from 8 to 7 +move 2 from 3 to 5 +move 1 from 7 to 1 +move 6 from 8 to 5 +move 2 from 2 to 9 +move 1 from 7 to 2 +move 2 from 9 to 2 +move 5 from 4 to 7 +move 3 from 2 to 7 +move 14 from 1 to 5 +move 2 from 4 to 7 +move 8 from 7 to 6 +move 1 from 1 to 5 +move 1 from 7 to 4 +move 1 from 7 to 5 +move 1 from 1 to 8 +move 12 from 3 to 4 +move 1 from 8 to 7 +move 3 from 4 to 1 +move 1 from 6 to 2 +move 8 from 5 to 2 +move 1 from 7 to 6 +move 1 from 1 to 7 +move 6 from 6 to 2 +move 1 from 1 to 2 +move 14 from 5 to 7 +move 1 from 6 to 4 +move 4 from 4 to 7 +move 1 from 1 to 6 +move 1 from 5 to 6 +move 2 from 3 to 1 +move 14 from 7 to 5 +move 10 from 4 to 7 +move 1 from 1 to 9 +move 1 from 5 to 9 +move 11 from 5 to 1 +move 6 from 7 to 6 +move 1 from 4 to 6 +move 1 from 3 to 7 +move 2 from 1 to 5 +move 13 from 2 to 1 +move 10 from 6 to 7 +move 4 from 5 to 2 +move 1 from 9 to 1 +move 1 from 3 to 6 +move 2 from 5 to 2 +move 1 from 9 to 3 +move 1 from 3 to 1 +move 21 from 7 to 5 +move 1 from 6 to 4 +move 4 from 5 to 1 +move 1 from 4 to 1 +move 6 from 2 to 3 +move 1 from 3 to 6 +move 1 from 3 to 8 +move 1 from 8 to 7 +move 1 from 7 to 3 +move 9 from 5 to 3 +move 24 from 1 to 4 +move 1 from 3 to 7 +move 11 from 3 to 8 +move 1 from 7 to 3 +move 1 from 2 to 4 +move 2 from 2 to 1 +move 2 from 3 to 5 +move 1 from 6 to 5 +move 10 from 4 to 6 +move 2 from 6 to 4 +move 5 from 1 to 2 +move 1 from 6 to 7 +move 8 from 8 to 6 +move 4 from 2 to 7 +move 8 from 6 to 7 +move 1 from 2 to 8 +move 1 from 8 to 3 +move 1 from 7 to 4 +move 3 from 4 to 1 +move 2 from 6 to 7 +move 4 from 1 to 9 +move 3 from 6 to 7 +move 10 from 7 to 4 +move 2 from 3 to 9 +move 2 from 6 to 9 +move 2 from 1 to 8 +move 2 from 9 to 5 +move 4 from 5 to 6 +move 3 from 8 to 1 +move 4 from 4 to 8 +move 5 from 8 to 4 +move 1 from 8 to 2 +move 5 from 5 to 9 +move 1 from 6 to 1 +move 2 from 1 to 7 +move 22 from 4 to 8 +move 4 from 8 to 7 +move 2 from 6 to 7 +move 1 from 2 to 6 +move 16 from 8 to 9 +move 3 from 7 to 4 +move 1 from 5 to 9 +move 2 from 6 to 7 +move 1 from 8 to 2 +move 1 from 2 to 3 +move 24 from 9 to 3 +move 1 from 1 to 7 +move 3 from 5 to 1 +move 4 from 4 to 6 +move 15 from 3 to 6 +move 18 from 6 to 2 +move 3 from 3 to 2 +move 4 from 1 to 6 +move 4 from 7 to 3 +move 1 from 3 to 9 +move 4 from 2 to 1 +move 1 from 8 to 7 +move 3 from 9 to 6 +move 1 from 9 to 3 +move 4 from 7 to 3 +move 2 from 4 to 2 +move 1 from 1 to 2 +move 7 from 3 to 5 +move 8 from 6 to 1 +move 1 from 9 to 2 +move 3 from 7 to 5 +move 1 from 4 to 8 +move 3 from 1 to 7 +move 5 from 7 to 6 +move 3 from 5 to 2 +move 3 from 7 to 3 +move 5 from 5 to 9 +move 5 from 3 to 6 +move 1 from 8 to 3 +move 5 from 9 to 7 +move 7 from 2 to 4 +move 11 from 2 to 7 +move 7 from 1 to 6 +move 1 from 1 to 9 +move 5 from 3 to 6 +move 5 from 2 to 1 +move 1 from 3 to 9 +move 1 from 3 to 7 +move 6 from 6 to 2 +move 10 from 6 to 7 +move 5 from 6 to 7 +move 28 from 7 to 8 +move 2 from 9 to 1 +move 1 from 6 to 3 +move 4 from 7 to 5 +move 1 from 3 to 6 +move 7 from 2 to 7 +move 6 from 7 to 3 +move 1 from 5 to 9 +move 1 from 6 to 2 +move 1 from 7 to 3 +move 1 from 9 to 1 +move 4 from 5 to 2 +move 5 from 3 to 5 +move 2 from 2 to 8 +move 4 from 4 to 7 +move 1 from 4 to 7 +move 2 from 3 to 6 +move 5 from 7 to 1 +move 2 from 5 to 8 +move 2 from 5 to 8 +move 2 from 5 to 3 +move 2 from 3 to 1 +move 2 from 6 to 7 +move 31 from 8 to 3 +move 2 from 8 to 5 +move 2 from 7 to 4 +move 7 from 1 to 4 +move 2 from 5 to 1 +move 3 from 2 to 8 +move 2 from 4 to 6 +move 3 from 1 to 2 +move 6 from 4 to 8 +move 1 from 1 to 8 +move 1 from 6 to 5 +move 11 from 8 to 9 +move 1 from 6 to 8 +move 1 from 4 to 1 +move 1 from 8 to 7 +move 1 from 5 to 8 +move 3 from 2 to 1 +move 2 from 4 to 3 +move 1 from 8 to 1 +move 7 from 3 to 6 +move 12 from 3 to 2 +move 1 from 7 to 9 +move 4 from 6 to 1 +move 1 from 6 to 3 +move 12 from 9 to 3 +move 1 from 6 to 4 +move 1 from 1 to 7 +move 1 from 4 to 1 +move 1 from 7 to 2 +move 1 from 6 to 5 +move 1 from 5 to 6 +move 5 from 3 to 1 +move 1 from 6 to 4 +move 7 from 2 to 1 +move 3 from 2 to 6 +move 1 from 4 to 5 +move 3 from 3 to 2 +move 4 from 2 to 8 +move 1 from 6 to 4 +move 1 from 4 to 9 +move 1 from 5 to 1 +move 11 from 1 to 5 +move 10 from 1 to 8 +move 2 from 6 to 4 +move 1 from 2 to 9 +move 1 from 2 to 4 +move 18 from 3 to 5 +move 4 from 1 to 4 +move 3 from 1 to 2 +move 14 from 8 to 5 +move 2 from 2 to 6 +move 1 from 3 to 2 +move 2 from 2 to 7 +move 3 from 4 to 1 +move 2 from 4 to 3 +move 2 from 3 to 4 +move 2 from 6 to 9 +move 1 from 7 to 1 +move 3 from 1 to 4 +move 4 from 9 to 7 +move 31 from 5 to 2 +move 25 from 2 to 4 +move 13 from 4 to 2 +move 10 from 2 to 3 +move 2 from 5 to 7 +move 5 from 2 to 9 +move 7 from 5 to 7 +move 5 from 7 to 4 +move 1 from 5 to 8 +move 2 from 7 to 3 +move 11 from 4 to 8 +move 1 from 7 to 3 +move 1 from 1 to 4 +move 2 from 5 to 3 +move 3 from 2 to 9 +move 8 from 9 to 6 +move 10 from 8 to 2 +move 5 from 3 to 2 +move 1 from 7 to 3 +move 3 from 7 to 3 +move 15 from 2 to 1 +move 11 from 1 to 3 +move 1 from 8 to 2 +move 8 from 6 to 5 +move 1 from 2 to 6 +move 1 from 6 to 1 +move 12 from 3 to 7 +move 1 from 2 to 9 +move 2 from 4 to 1 +move 3 from 1 to 8 +move 1 from 8 to 7 +move 3 from 3 to 4 +move 1 from 4 to 7 +move 15 from 7 to 9 +move 1 from 7 to 5 +move 4 from 1 to 8 +move 6 from 8 to 6 +move 1 from 6 to 2 +move 5 from 5 to 1 +move 2 from 6 to 8 +move 1 from 2 to 7 +move 1 from 8 to 2 +move 1 from 7 to 1 +move 1 from 5 to 8 +move 6 from 3 to 1 +move 4 from 3 to 8 +move 7 from 8 to 5 +move 1 from 2 to 4 +move 2 from 4 to 2 +move 3 from 6 to 4 +move 5 from 9 to 3 +move 4 from 1 to 4 +move 10 from 5 to 9 +move 8 from 1 to 7 +move 1 from 2 to 1 +move 1 from 1 to 9 +move 20 from 9 to 2 +move 12 from 2 to 3 +move 17 from 4 to 3 +move 6 from 7 to 2 +move 5 from 3 to 8 +move 20 from 3 to 5 +move 2 from 9 to 4 +move 3 from 3 to 1 +move 1 from 7 to 1 +move 6 from 3 to 6 +move 4 from 2 to 3 +move 4 from 5 to 3 +move 1 from 1 to 9 +move 6 from 6 to 1 +move 3 from 8 to 4 +move 1 from 9 to 8 +move 2 from 2 to 1 +move 3 from 3 to 2 +move 1 from 3 to 6 +move 1 from 7 to 4 +move 3 from 3 to 6 +move 6 from 1 to 5 +move 9 from 2 to 4 +move 3 from 2 to 5 +move 2 from 6 to 5 +move 16 from 4 to 8 +move 18 from 8 to 6 +move 1 from 4 to 5 +move 2 from 6 to 7 +move 4 from 1 to 7 +move 22 from 5 to 6 +move 1 from 4 to 9 +move 4 from 7 to 6 +move 11 from 6 to 5 +move 9 from 5 to 2 +move 2 from 2 to 3 +move 2 from 7 to 2 +move 1 from 1 to 7 +move 9 from 6 to 2 +move 1 from 5 to 1 +move 1 from 8 to 9 +move 18 from 6 to 8 +move 1 from 7 to 4 +move 4 from 5 to 1 +move 2 from 5 to 2 +move 2 from 2 to 5 +move 1 from 9 to 5 +move 1 from 5 to 9 +move 1 from 9 to 1 +move 1 from 9 to 2 +move 1 from 4 to 8 +move 4 from 1 to 4 +move 2 from 6 to 5 +move 1 from 1 to 9 +move 3 from 6 to 7 +move 1 from 6 to 9 +move 1 from 9 to 8 +move 2 from 5 to 9 +move 3 from 3 to 5 +move 7 from 2 to 3 +move 1 from 1 to 3 +move 2 from 5 to 9 +move 1 from 5 to 7 +move 10 from 8 to 3 +move 10 from 8 to 9 +move 3 from 4 to 3 +move 9 from 2 to 1 +move 4 from 9 to 6 +move 5 from 1 to 9 +move 2 from 5 to 9 +move 1 from 6 to 4 +move 4 from 7 to 2 +move 7 from 2 to 9 +move 3 from 6 to 8 +move 1 from 1 to 3 +move 2 from 8 to 5 +move 1 from 8 to 1 +move 18 from 3 to 6 +move 15 from 9 to 2 +move 8 from 9 to 1 +move 2 from 9 to 2 +move 2 from 4 to 9 +move 2 from 9 to 7 +move 12 from 6 to 3 +move 7 from 1 to 7 +move 12 from 2 to 5 +move 7 from 3 to 2 +move 4 from 3 to 4 +move 2 from 7 to 6 +move 7 from 7 to 8 +move 1 from 4 to 2 +move 4 from 1 to 8 +move 5 from 3 to 1 +move 9 from 8 to 3 +move 1 from 8 to 7 +move 2 from 1 to 2 +move 4 from 6 to 7 +move 11 from 2 to 5 +move 2 from 4 to 6 +move 1 from 8 to 2 +move 7 from 3 to 2 +move 1 from 2 to 4 +move 4 from 6 to 1 +move 7 from 5 to 8 +move 2 from 3 to 1 +move 7 from 2 to 3 +move 6 from 5 to 1 +move 1 from 4 to 2 +move 8 from 1 to 6 +move 3 from 2 to 9 diff --git a/5.test b/5.test new file mode 100644 index 0000000..edacdb9 --- /dev/null +++ b/5.test @@ -0,0 +1,10 @@ + [D] +[N] [C] +[Z] [M] [P] + 1 2 3 + +move 1 from 2 to 1 +move 3 from 1 to 3 +move 2 from 2 to 1 +move 1 from 1 to 2 +