aboutsummaryrefslogtreecommitdiff
path: root/execs/Day12.hs
blob: 130e5393dc2af29ce744a3b22c974135d633c924 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
module Main where

import           Utils

parseLine s = (head s, read (tail s) :: Float)

moves ls = sum $ map (round . abs) [fx, fy]
    where (fx, fy, _) = foldl fn (0.0, 0.0, 0.0) ls
          fn (x, y, rot) ('N', v) = (x, y + v, rot)
          fn (x, y, rot) ('E', v) = (x + v, y, rot)
          fn (x, y, rot) ('W', v) = (x - v, y, rot)
          fn (x, y, rot) ('S', v) = (x, y - v, rot)
          fn (x, y, rot) ('R', v) = (x, y, rot - v)
          fn (x, y, rot) ('L', v) = (x, y, rot + v)
          fn (x, y, rot) ('F', v) = (nx, ny, rot)
              where nx = x + v * cos (pi * rot / 180)
                    ny = y + v * sin (pi * rot / 180)

waypoint (sx, sy) ls = sum $ map (round . abs) [fx, fy]
    where (fx, fy, _, _) = foldl fn (0.0, 0.0, sx, sy) ls
          fn (x, y, wx, wy) ('N', v) = (x, y, wx, wy+v)
          fn (x, y, wx, wy) ('E', v) = (x, y, wx+v, wy)
          fn (x, y, wx, wy) ('W', v) = (x, y, wx-v, wy)
          fn (x, y, wx, wy) ('S', v) = (x, y, wx, wy-v)
          fn (x, y, wx, wy) ('R', v) = (x, y, nwx, nwy)
              where (nwx, nwy) = rotate (wx, wy) (negate v)
          fn (x, y, wx, wy) ('L', v) = (x, y, nwx, nwy)
              where (nwx, nwy) = rotate (wx, wy) v
          fn (x, y, wx, wy) ('F', v) = (x+v*wx, y+v*wy, wx, wy)

main :: IO ()
main = do
    n <- map parseLine . lines <$> readFile "input/12"
    print $ moves n
    print $ waypoint (10.0, 1.0) n