Gattix/AoC
3
0
Fork 0
Code Issues 1 Pull Requests Releases Wiki Activity

Day 9: Some more cosmetics

This commit is contained in:
shu 2019-12-09 17:35:34 +01:00
parent bef16465d7
commit 90a65ce187
1 changed files with 34 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
2019/day9/day9.hs
View File

@ -1,6 +1,7 @@
import Data.List.Split
import qualified Data.List as L
import Data.Vector as V
import Data.List as L
import Data.Vector as V hiding ((++), concatMap, map, elem, last, head, length, splitAt, reverse)
import qualified Data.Vector as V ((++), last)
import Data.Char
import Control.DeepSeq as DeepSeq
@ -12,24 +13,27 @@ type TuringMachine = (Tape,Integer,Integer)
main = do
content <- readFile "input"
let tape = fromList $ L.concatMap (L.map read . splitOn ",") (lines content)
let (tm,_,out,_) = execSteps ((tapePreprocess tape,0,0),[2],[],Continue)
print $ L.reverse out
let tape = fromList $ concatMap (map read . splitOn ",") (lines content)
let run x = execSteps ((tapePreprocess tape,0,0),[x],[],Continue)
let (_,_,out1,_) = run 1
let (_,_,out2,_) = run 2
print $ L.concat ["Part 1: " ++ show a ++ ", Part 2: " ++ show b | a<-out1,b<-out2]
tapePreprocess :: TapeSection -> TapeSection
tapePreprocess t = (V.++) t $ V.replicate 150 0
tapePreprocess t = (V.++) t $ V.replicate 105 0
opLength :: Integer -> Integer
opLength x
| n`L.elem`"1278"=4
| n`L.elem`"56"=3
| n`L.elem`"349"=2
| n`elem`"1278"=4
| n`elem`"56"=3
| n`elem`"349"=2
| otherwise=1
where n = L.last $ show x
where n = last $ show x
parseModes :: String -> [Mode]
parseModes m = L.replicate (3 - L.length l) Position L.++ l
where l = L.map (toEnum . digitToInt) m
parseModes m = L.replicate (3 - length l) Position ++ l
where l = map (toEnum . digitToInt) m
paramChange :: [Mode] -> Integer -> TapeSection -> Tape -> TapeSection
paramChange m rbase opvec t = imap f (V.tail opvec)
@ -40,31 +44,30 @@ paramChange m rbase opvec t = imap f (V.tail opvec)
getOpModes :: TapeSection -> (String, [Mode])
getOpModes opvec = (op_dedup,parsed_modes)
where (op,modes) = L.splitAt 2 $ L.reverse $ show $ opvec ! 0
parsed_modes = L.reverse $ parseModes $ L.reverse modes
op_dedup = if L.last op == '0' then [L.head op] else op
where (op,modes) = splitAt 2 $ reverse $ show $ opvec ! 0
parsed_modes = reverse $ parseModes $ reverse modes
op_dedup = if last op == '0' then [head op] else op
step :: TapeSection -> (TuringMachine, [Integer], [Integer]) -> (TuringMachine, OutAction, [Integer], [Integer])
step opvec ((t,p,rbase),input,val) = case op of
"1" -> (tm_binop (+),Continue,input,val)
"2" -> (tm_binop (*),Continue,input,val)
"3" -> (new_tm t $ L.head input,Continue,L.tail input,val)
"4" -> ((t,p,rbase),Output,input,V.last params:val)
"5" -> ((t, if params ! 0/=0 then params ! 1 else p,rbase),Continue,input,val)
"6" -> ((t, if params ! 0==0 then params ! 1 else p,rbase),Continue,input,val)
"7" -> (new_tm t (if (params ! 0)<(params ! 1)
then 1 else 0),Continue,input,val)
"8" -> (new_tm t (if (params ! 0)==(params ! 1)
then 1 else 0),Continue,input,val)
"9" -> ((t,p,rbase + (params ! 0)),Continue,input,val)
"99" -> ((t,p,rbase),Halt,input,val)
step opvec ((t,p,rbase),input,output) = case op of
"1" -> (tm_binop (+),Continue,input,output)
"2" -> (tm_binop (*),Continue,input,output)
"3" -> (new_tm t $ head input,Continue,L.tail input,output)
"4" -> ((t,p,rbase),Output,input,V.last params:output)
"5" -> ((t, if params ! 0/=0 then params ! 1 else p,rbase),Continue,input,output)
"6" -> ((t, if params ! 0==0 then params ! 1 else p,rbase),Continue,input,output)
"7" -> (tm_binop (\x y->if x<y then 1 else 0),Continue,input,output)
"8" -> (tm_binop (\x y->if x==y then 1 else 0),Continue,input,output)
"9" -> ((t,p,rbase + (params ! 0)),Continue,input,output)
"99" -> ((t,p,rbase),Halt,input,output)
where (op,m) = getOpModes opvec
params = paramChange m rbase opvec t
tm_binop x = new_tm t ((params ! 0) `x` (params ! 1))
--without the following DeepSeq call, thunks build up eternally
--and the vectors won’t be garbage collected (>4GB RAM usage)
{-without the following DeepSeq call, thunks build up eternally
and the vectors won’t be garbage collected: >4GB RAM usage,
god knows how much with large vectors, now ~20mB-}
new_tm t x = DeepSeq.force (t // [(target, x)],p,rbase)
target = fromInteger $ case m !! (L.length params -1) of
target = fromInteger $ case m !! (length params -1) of
Relative -> V.last opvec + rbase
_ -> V.last opvec