diff options
Diffstat (limited to '2019')
| -rw-r--r-- | 2019/17.hs | 142 | ||||
| -rw-r--r-- | 2019/17.in | 1 |
2 files changed, 143 insertions, 0 deletions
diff --git a/2019/17.hs b/2019/17.hs new file mode 100644 index 0000000..067305b --- /dev/null +++ b/2019/17.hs @@ -0,0 +1,142 @@ +{-# LANGUAGE MultiWayIf #-} +module Main where + +import Control.Monad +import Data.Char +import Data.List +import Data.Maybe +import qualified Data.Map.Strict as Map + +import Input +import IntCode + + +blockBy :: Int -> [a] -> [[a]] +blockBy n l = + let (pre, post) = splitAt n l + in if null post then [pre] else pre : blockBy n post + +weightLimit :: Int +weightLimit = 20 + +numFunctions :: Int +numFunctions = 3 + +data Move = F Int | L | R deriving (Show, Eq) +type Path = [Move] + +weight :: Path -> Int +weight l = length l - 1 + sum (map moveWeight l) + +moveWeight :: Move -> Int +moveWeight (F n) = numberWeight n +moveWeight _ = 1 + +numberWeight :: Int -> Int +numberWeight n | n < 10 = 1 + | otherwise = 1 + numberWeight (n `quot` 10) + +limitedPrefixes :: Path -> [(Path, Path)] +limitedPrefixes moves = + let prefws = tail (scanl (+) (-1) (map (succ . moveWeight) moves)) + maxlen = length (takeWhile (<= weightLimit) prefws) + in [splitAt i moves | i <- [1..maxlen]] + +data State = State { pathFuncs :: [Path], pathRemain :: Path } deriving (Show) + +compress' :: State -> [([Int], State)] +compress' state@(State _ []) = return ([], state) +compress' (State funcs remain) = + let match idx func = do + guard $ take (length func) remain == func + return ([idx], State funcs (drop (length func) remain)) + matches = concatMap (uncurry match) (zip [0..] funcs) + newFunc prefix rest = do + let newidx = length funcs + guard $ newidx < numFunctions + return ([newidx], State (funcs ++ [prefix]) rest) + newFuncs = concatMap (uncurry newFunc) (limitedPrefixes remain) + options = matches ++ newFuncs + developOption (yet, state) = do + (yet', state') <- compress' state + return (yet ++ yet', state') + in concatMap developOption options + +compress :: Path -> [([Int], [Path])] +compress path = do + (calls, State funcs _) <- compress' (State [] path) + let mainWeight = length calls - 1 + sum (map numberWeight calls) + guard $ mainWeight <= weightLimit + let restored = concatMap (funcs !!) calls + when (restored /= path) (error "kaas!") + return (calls, funcs) + +type Pos = (Int, Int) +type Dir = (Int, Int) + +add :: Pos -> Dir -> Pos +add (x, y) (dx, dy) = (x + dx, y + dy) + +rotR, rotL :: Dir -> Dir +rotR (dx, dy) = (-dy, dx) +rotL (dx, dy) = (dy, -dx) + +fourDirections :: [Dir] +fourDirections = [(-1,0), (1,0), (0,-1), (0,1)] + +findPath :: Map.Map Pos Char -> Pos -> Dir -> Path +findPath bd pos dir = + let cands = [([F 1], add pos dir, dir), + ([R, F 1], add pos (rotR dir), rotR dir), + ([L, F 1], add pos (rotL dir), rotL dir)] + cands' = filter (\(_, p, _) -> Map.lookup p bd == Just '#') cands + in case cands' of + (moves, pos', dir') : _ -> moves ++ findPath bd pos' dir' + _ -> [] + +simplifyPath :: Path -> Path +simplifyPath (F n : F m : rest) = simplifyPath (F (n + m) : rest) +simplifyPath (m : rest) = m : simplifyPath rest +simplifyPath [] = [] + +main :: IO () +main = do + program <- parse . head <$> getInput 17 + let bd = let output = map (chr . fromIntegral) (snd (run program [])) + nlidx = fromJust (findIndex (== '\n') output) + in blockBy nlidx (filter (/= '\n') output) + w = length (head bd) + h = length bd + bdmap = Map.fromList [((x, y), c) | (y, row) <- zip [0..] bd, (x, c) <- zip [0..] row] + + -- mapM_ print bd + + let inters = [(x, y) + | y <- [1..h-2] + , x <- [1..w-2] + , all (== '#') [bdmap Map.! (x+dx, y+dy) | (dx, dy) <- (0,0) : fourDirections]] + -- print inters + + -- putStr (unlines [concat [if (x, y) `elem` inters + -- then "\x1B[41;1m" ++ cell : "\x1B[0m" + -- else [cell] + -- | (x, cell) <- zip [0..] row] + -- | (y, row) <- zip [0..] bd]) + + print (sum (map (uncurry (*)) inters)) + + let startpos = fst (head (filter ((`elem` "^>v<") . snd) (Map.assocs bdmap))) + startdir = fromJust (lookup (bdmap Map.! startpos) (zip "<>^v" fourDirections)) + path = simplifyPath (findPath bdmap startpos startdir) + compressed = compress path + (mainroutine, funcs) = head compressed + -- mapM_ print compressed + + let program' = 2 : tail program + moveInput = map (fromIntegral . ord) . unlines . map (intercalate ",") $ + [map (pure . ("ABC" !!)) mainroutine] + ++ [[case m of {F n -> show n; R -> "R"; L -> "L"} | m <- func] + | func <- funcs] + ++ [["n"]] + + print (last (snd (run program' moveInput))) diff --git a/2019/17.in b/2019/17.in new file mode 100644 index 0000000..0fed63c --- /dev/null +++ b/2019/17.in @@ -0,0 +1 @@ +1,330,331,332,109,3664,1102,1,1182,15,1101,0,1455,24,1001,0,0,570,1006,570,36,102,1,571,0,1001,570,-1,570,1001,24,1,24,1105,1,18,1008,571,0,571,1001,15,1,15,1008,15,1455,570,1006,570,14,21101,58,0,0,1105,1,786,1006,332,62,99,21102,333,1,1,21102,1,73,0,1105,1,579,1102,1,0,572,1101,0,0,573,3,574,101,1,573,573,1007,574,65,570,1005,570,151,107,67,574,570,1005,570,151,1001,574,-64,574,1002,574,-1,574,1001,572,1,572,1007,572,11,570,1006,570,165,101,1182,572,127,102,1,574,0,3,574,101,1,573,573,1008,574,10,570,1005,570,189,1008,574,44,570,1006,570,158,1105,1,81,21101,340,0,1,1105,1,177,21102,1,477,1,1106,0,177,21101,514,0,1,21102,1,176,0,1105,1,579,99,21102,1,184,0,1106,0,579,4,574,104,10,99,1007,573,22,570,1006,570,165,102,1,572,1182,21102,375,1,1,21101,211,0,0,1106,0,579,21101,1182,11,1,21102,222,1,0,1105,1,979,21102,1,388,1,21101,233,0,0,1105,1,579,21101,1182,22,1,21102,1,244,0,1105,1,979,21102,401,1,1,21101,0,255,0,1106,0,579,21101,1182,33,1,21101,266,0,0,1106,0,979,21101,414,0,1,21102,277,1,0,1105,1,579,3,575,1008,575,89,570,1008,575,121,575,1,575,570,575,3,574,1008,574,10,570,1006,570,291,104,10,21101,0,1182,1,21102,313,1,0,1105,1,622,1005,575,327,1102,1,1,575,21102,1,327,0,1105,1,786,4,438,99,0,1,1,6,77,97,105,110,58,10,33,10,69,120,112,101,99,116,101,100,32,102,117,110,99,116,105,111,110,32,110,97,109,101,32,98,117,116,32,103,111,116,58,32,0,12,70,117,110,99,116,105,111,110,32,65,58,10,12,70,117,110,99,116,105,111,110,32,66,58,10,12,70,117,110,99,116,105,111,110,32,67,58,10,23,67,111,110,116,105,110,117,111,117,115,32,118,105,100,101,111,32,102,101,101,100,63,10,0,37,10,69,120,112,101,99,116,101,100,32,82,44,32,76,44,32,111,114,32,100,105,115,116,97,110,99,101,32,98,117,116,32,103,111,116,58,32,36,10,69,120,112,101,99,116,101,100,32,99,111,109,109,97,32,111,114,32,110,101,119,108,105,110,101,32,98,117,116,32,103,111,116,58,32,43,10,68,101,102,105,110,105,116,105,111,110,115,32,109,97,121,32,98,101,32,97,116,32,109,111,115,116,32,50,48,32,99,104,97,114,97,99,116,101,114,115,33,10,94,62,118,60,0,1,0,-1,-1,0,1,0,0,0,0,0,0,1,46,22,0,109,4,2102,1,-3,587,20102,1,0,-1,22101,1,-3,-3,21102,1,0,-2,2208,-2,-1,570,1005,570,617,2201,-3,-2,609,4,0,21201,-2,1,-2,1106,0,597,109,-4,2105,1,0,109,5,2102,1,-4,629,21002,0,1,-2,22101,1,-4,-4,21101,0,0,-3,2208,-3,-2,570,1005,570,781,2201,-4,-3,652,21001,0,0,-1,1208,-1,-4,570,1005,570,709,1208,-1,-5,570,1005,570,734,1207,-1,0,570,1005,570,759,1206,-1,774,1001,578,562,684,1,0,576,576,1001,578,566,692,1,0,577,577,21101,702,0,0,1105,1,786,21201,-1,-1,-1,1105,1,676,1001,578,1,578,1008,578,4,570,1006,570,724,1001,578,-4,578,21102,731,1,0,1106,0,786,1106,0,774,1001,578,-1,578,1008,578,-1,570,1006,570,749,1001,578,4,578,21102,1,756,0,1105,1,786,1105,1,774,21202,-1,-11,1,22101,1182,1,1,21102,774,1,0,1106,0,622,21201,-3,1,-3,1105,1,640,109,-5,2106,0,0,109,7,1005,575,802,20101,0,576,-6,21002,577,1,-5,1105,1,814,21101,0,0,-1,21101,0,0,-5,21102,0,1,-6,20208,-6,576,-2,208,-5,577,570,22002,570,-2,-2,21202,-5,47,-3,22201,-6,-3,-3,22101,1455,-3,-3,2102,1,-3,843,1005,0,863,21202,-2,42,-4,22101,46,-4,-4,1206,-2,924,21102,1,1,-1,1106,0,924,1205,-2,873,21101,0,35,-4,1105,1,924,1202,-3,1,878,1008,0,1,570,1006,570,916,1001,374,1,374,2101,0,-3,895,1101,2,0,0,1201,-3,0,902,1001,438,0,438,2202,-6,-5,570,1,570,374,570,1,570,438,438,1001,578,558,921,21001,0,0,-4,1006,575,959,204,-4,22101,1,-6,-6,1208,-6,47,570,1006,570,814,104,10,22101,1,-5,-5,1208,-5,47,570,1006,570,810,104,10,1206,-1,974,99,1206,-1,974,1101,0,1,575,21101,0,973,0,1105,1,786,99,109,-7,2105,1,0,109,6,21102,0,1,-4,21101,0,0,-3,203,-2,22101,1,-3,-3,21208,-2,82,-1,1205,-1,1030,21208,-2,76,-1,1205,-1,1037,21207,-2,48,-1,1205,-1,1124,22107,57,-2,-1,1205,-1,1124,21201,-2,-48,-2,1105,1,1041,21101,0,-4,-2,1105,1,1041,21101,-5,0,-2,21201,-4,1,-4,21207,-4,11,-1,1206,-1,1138,2201,-5,-4,1059,2102,1,-2,0,203,-2,22101,1,-3,-3,21207,-2,48,-1,1205,-1,1107,22107,57,-2,-1,1205,-1,1107,21201,-2,-48,-2,2201,-5,-4,1090,20102,10,0,-1,22201,-2,-1,-2,2201,-5,-4,1103,2101,0,-2,0,1105,1,1060,21208,-2,10,-1,1205,-1,1162,21208,-2,44,-1,1206,-1,1131,1105,1,989,21101,0,439,1,1105,1,1150,21102,1,477,1,1106,0,1150,21102,1,514,1,21102,1,1149,0,1105,1,579,99,21101,0,1157,0,1106,0,579,204,-2,104,10,99,21207,-3,22,-1,1206,-1,1138,2102,1,-5,1176,2101,0,-4,0,109,-6,2106,0,0,14,5,42,1,3,1,42,1,3,1,42,1,3,1,42,1,3,1,42,1,3,1,42,1,3,1,5,13,24,1,3,1,5,1,11,1,24,1,1,5,3,1,1,13,22,1,1,1,1,1,1,1,3,1,1,1,9,1,1,1,22,1,1,1,1,13,5,1,1,1,22,1,1,1,3,1,3,1,1,1,3,1,5,1,1,1,22,7,3,1,1,1,1,11,24,1,7,1,1,1,1,1,1,1,5,1,18,9,7,1,1,5,5,5,14,1,15,1,3,1,11,1,14,1,15,1,3,1,11,1,14,1,15,1,3,1,11,1,6,9,5,11,3,1,11,1,6,1,13,1,13,1,11,1,6,1,13,1,9,7,9,1,6,1,13,1,9,1,3,1,1,1,9,1,6,1,13,13,1,1,1,1,9,8,23,1,1,1,1,1,1,1,16,1,23,5,1,1,16,1,25,1,3,1,16,1,25,1,3,1,16,1,25,1,3,1,16,1,25,1,3,1,16,1,25,1,3,1,16,7,19,1,3,1,22,1,19,1,3,1,22,1,19,5,22,1,46,1,46,1,46,1,46,1,46,5,46,1,46,1,46,1,46,1,46,1,46,1,46,1,46,13,24 |