aboutsummaryrefslogtreecommitdiff
path: root/2020/08a.hs
diff options
context:
space:
mode:
Diffstat (limited to '2020/08a.hs')
-rw-r--r--2020/08a.hs34
1 files changed, 34 insertions, 0 deletions
diff --git a/2020/08a.hs b/2020/08a.hs
new file mode 100644
index 0000000..846c3df
--- /dev/null
+++ b/2020/08a.hs
@@ -0,0 +1,34 @@
+module Day8A where
+
+import System.Environment (getArgs)
+
+data Instruction = Nop | Acc Int | Jmp Int
+ deriving (Eq, Show, Ord)
+
+readSigned :: String -> Int
+readSigned ('+':xs) = read xs
+readSigned xs = read xs
+
+parseInstruction :: String -> Instruction
+parseInstruction xs | ins == "nop" = Nop
+ | ins == "acc" = Acc num
+ | ins == "jmp" = Jmp num
+ where ins = take 3 xs
+ num = readSigned $ drop 4 xs
+
+execUntilLoop :: [Instruction] -> [Int] -> Int -> Int -> Int
+execUntilLoop is vs ip ax | ip `elem` vs = ax
+ | otherwise = case ins of Nop -> execUntilLoop is vs' (ip + 1) ax
+ Acc x -> execUntilLoop is vs' (ip + 1) (ax + x)
+ Jmp x -> execUntilLoop is vs' (ip + x) ax
+ where ins = is!!ip
+ vs' = ip : vs
+
+main :: IO ()
+main = do
+ args <- getArgs;
+ content <- readFile $ head args;
+ let l = lines content;
+ let is = map parseInstruction l;
+ print $ execUntilLoop is [] 0 0;
+ return (); \ No newline at end of file