Garmelon/advent-of-code
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[hs] Clean up 2020_12

Browse source
This commit is contained in:
Joscha 2020-12-12 12:14:30 +00:00
parent 92f43dea77
commit 559659cd73
1 changed files with 13 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

34
hs/src/Aoc/Y2020/D12.hs
View file

@ -10,16 +10,12 @@ import Aoc.Day
import Aoc.Parse import Aoc.Parse
data Pos = Pos Int Int data Pos = Pos Int Int
deriving (Show)
data Rot = RLeft | RRight | RFlip data Rot = RLeft | RRight | RFlip
deriving (Show)
data Move data Move
= MTranslate Pos Int = MTranslate Pos Int
| MForward Int | MForward Int
| MRotate Rot | MRotate Rot
deriving (Show)
parser :: Parser [Move] parser :: Parser [Move]
parser = manyLines parser = manyLines
@ -51,33 +47,29 @@ rotate RRight (Pos x y) = Pos (-y) x
rotate RLeft (Pos x y) = Pos y (-x) rotate RLeft (Pos x y) = Pos y (-x)
rotate RFlip (Pos x y) = Pos (-x) (-y) rotate RFlip (Pos x y) = Pos (-x) (-y)
data State1 = State1 Pos Pos -- Ship, direction data State = State Pos Pos -- Ship, direction/waypoint
deriving (Show)
step1 :: Move -> State1 -> State1 step1 :: Move -> State -> State
step1 (MTranslate dir steps) (State1 spos sdir) = State1 (add spos $ mul steps dir) sdir step1 (MTranslate dir steps) (State spos sdir) = State (add spos $ mul steps dir) sdir
step1 (MForward steps) (State1 spos sdir) = State1 (add spos $ mul steps sdir) sdir step1 (MForward steps) (State spos sdir) = State (add spos $ mul steps sdir) sdir
step1 (MRotate rot) (State1 spos sdir) = State1 spos (rotate rot sdir) step1 (MRotate rot) (State spos sdir) = State spos (rotate rot sdir)
data State2 = State2 Pos Pos -- Ship, waypoint step2 :: Move -> State -> State
deriving (Show) step2 (MTranslate dir steps) (State spos wp) = State spos (add wp $ mul steps dir)
step2 (MForward steps) (State spos wp) = State (add spos $ mul steps wp) wp
step2 :: Move -> State2 -> State2 step2 (MRotate rot) (State spos wp) = State spos (rotate rot wp)
step2 (MTranslate dir steps) (State2 spos wp) = State2 spos (add wp $ mul steps dir)
step2 (MForward steps) (State2 spos wp) = State2 (add spos $ mul steps wp) wp
step2 (MRotate rot) (State2 spos wp) = State2 spos (rotate rot wp)
solver :: [Move] -> IO () solver :: [Move] -> IO ()
solver moves = do solver moves = do
putStrLn ">> Part 1" putStrLn ">> Part 1"
let initialState1 = State1 (Pos 0 0) (Pos 1 0) let initialState1 = State (Pos 0 0) (Pos 1 0)
(State1 pos1 _) = foldl' (flip step1) initialState1 moves (State pos1 _) = foldl' (flip step1) initialState1 moves
print $ manhattan pos1 print $ manhattan pos1
putStrLn "" putStrLn ""
putStrLn ">> Part 2" putStrLn ">> Part 2"
let initialState2 = State2 (Pos 0 0) (Pos 10 (-1)) let initialState2 = State (Pos 0 0) (Pos 10 (-1))
(State2 pos2 _) = foldl' (flip step2) initialState2 moves (State pos2 _) = foldl' (flip step2) initialState2 moves
print $ manhattan pos2 print $ manhattan pos2
day :: Day day :: Day