day_12.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "day-12.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "authorship_tag": "ABX9TyNYgS2b3XSaChyDmhu2xie/",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/robinnewhouse/advent-of-code/blob/main/day_12.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "eWb7Ccia78cr",
        "outputId": "c49c09a4-12f2-4388-b8db-51bcf977c9f3"
      },
      "source": [
        "# part 1\n",
        "import numpy as np\n",
        "\n",
        "instructions = [(x[0], int(x[1:])) for x in input_string.splitlines()]\n",
        "\n",
        "# ship location\n",
        "x = 0\n",
        "y = 0\n",
        "d = 0 # direction\n",
        "\n",
        "for ins, val in instructions:\n",
        "  # cardinal\n",
        "  if ins == 'N': y += val\n",
        "  if ins == 'S': y -= val\n",
        "  if ins == 'E': x += val\n",
        "  if ins == 'W': x -= val\n",
        "  # forward\n",
        "  if ins == 'F': \n",
        "    x += val*np.cos(np.deg2rad(d))\n",
        "    y += val*np.sin(np.deg2rad(d))\n",
        "  # turn\n",
        "  if ins == 'R': d = d - val\n",
        "  if ins == 'L': d = d + val\n",
        "  d = d % 360\n",
        "    \n",
        "print(x, y, d)\n",
        "print(\"Manhattan distance:\", abs(x) + abs(y))\n",
        "print(\"closest integer:\", round(abs(x) + abs(y)))"
      ],
      "execution_count": 2,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "1423.0 -634.0 180\n",
            "Manhattan distance: 2057.0\n",
            "closest integer: 2057.0\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "JAXUCHaHDZsp",
        "outputId": "ce14310f-8db2-4970-ce62-c35200b1586a"
      },
      "source": [
        "# part 2\n",
        "import numpy as np\n",
        "\n",
        "instructions = [(x[0], int(x[1:])) for x in input_string.splitlines()]\n",
        "\n",
        "# ship location\n",
        "x = 0\n",
        "y = 0\n",
        "# waypoint location\n",
        "wx = 10\n",
        "wy = 1\n",
        "d = 0 # direction\n",
        "\n",
        "def rotate(point, about, d):\n",
        "  θ = np.deg2rad(d)\n",
        "  R = np.array([[np.cos(θ), -np.sin(θ)],\n",
        "                [np.sin(θ),  np.cos(θ)]])\n",
        "  \n",
        "  return np.add(R.dot(\n",
        "                  np.subtract(point, about)),\n",
        "                about)\n",
        "\n",
        "# print('ship:', x, y)\n",
        "# print('waypoint:', wx, wy)\n",
        "# print()\n",
        "\n",
        "for ins, val in instructions:\n",
        "  # cardinal\n",
        "  if ins == 'N': wy += val\n",
        "  if ins == 'S': wy -= val\n",
        "  if ins == 'E': wx += val\n",
        "  if ins == 'W': wx -= val\n",
        "  # turn\n",
        "  if ins == 'R': \n",
        "    wx, wy = rotate(point=(wx, wy), about=(0,0), d=-val)\n",
        "  if ins == 'L': \n",
        "    wx, wy = rotate(point=(wx, wy), about=(0,0), d=val)\n",
        "  # forward\n",
        "  if ins == 'F':\n",
        "    x, y = x + val*wx, y + val*wy\n",
        "\n",
        "  # print('ship:', x, y)\n",
        "  # print('waypoint:', wx, wy)\n",
        "  # print()\n",
        "\n",
        "print(\"Manhattan distance:\", abs(x) + abs(y))\n",
        "print(\"closest integer:\", round(abs(x) + abs(y)))"
      ],
      "execution_count": 71,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "Manhattan distance: 71503.99999999997\n",
            "closest integer: 71504.0\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "WmaJOX6R8gc6"
      },
      "source": [
        "test_string = '''F10\n",
        "N3\n",
        "F7\n",
        "R90\n",
        "F11\n",
        "'''\n",
        "\n",
        "input_string='''E2\n",
        "F24\n",
        "F60\n",
        "N3\n",
        "E1\n",
        "R180\n",
        "F14\n",
        "L90\n",
        "N4\n",
        "W3\n",
        "F81\n",
        "R90\n",
        "F43\n",
        "R90\n",
        "F40\n",
        "S2\n",
        "F56\n",
        "L90\n",
        "E1\n",
        "S1\n",
        "R90\n",
        "F66\n",
        "L180\n",
        "W2\n",
        "F84\n",
        "R90\n",
        "E4\n",
        "F32\n",
        "E5\n",
        "R90\n",
        "F44\n",
        "E1\n",
        "R90\n",
        "E4\n",
        "L90\n",
        "N1\n",
        "W2\n",
        "R90\n",
        "E5\n",
        "L90\n",
        "F86\n",
        "L180\n",
        "S2\n",
        "R180\n",
        "E2\n",
        "F22\n",
        "R180\n",
        "F42\n",
        "L180\n",
        "S4\n",
        "W4\n",
        "N5\n",
        "R270\n",
        "F23\n",
        "N3\n",
        "F18\n",
        "E2\n",
        "N2\n",
        "S3\n",
        "W5\n",
        "R90\n",
        "E4\n",
        "S1\n",
        "R180\n",
        "S1\n",
        "F69\n",
        "W2\n",
        "F55\n",
        "E5\n",
        "S3\n",
        "E4\n",
        "N1\n",
        "R180\n",
        "E1\n",
        "S5\n",
        "W1\n",
        "S5\n",
        "L270\n",
        "F17\n",
        "R180\n",
        "N5\n",
        "F6\n",
        "S3\n",
        "R180\n",
        "F75\n",
        "S3\n",
        "F90\n",
        "W3\n",
        "F98\n",
        "N2\n",
        "F54\n",
        "E5\n",
        "S5\n",
        "F4\n",
        "E5\n",
        "R90\n",
        "N4\n",
        "L90\n",
        "E4\n",
        "N3\n",
        "F75\n",
        "E3\n",
        "S2\n",
        "F89\n",
        "E1\n",
        "R180\n",
        "S3\n",
        "E2\n",
        "R270\n",
        "W4\n",
        "L90\n",
        "N5\n",
        "E4\n",
        "W3\n",
        "N1\n",
        "F37\n",
        "L90\n",
        "N5\n",
        "E3\n",
        "N3\n",
        "L90\n",
        "W1\n",
        "R90\n",
        "F25\n",
        "N2\n",
        "R90\n",
        "E4\n",
        "F81\n",
        "N5\n",
        "L90\n",
        "F47\n",
        "E1\n",
        "F86\n",
        "N5\n",
        "W1\n",
        "N2\n",
        "F35\n",
        "L90\n",
        "W4\n",
        "F35\n",
        "R90\n",
        "W3\n",
        "S3\n",
        "W1\n",
        "S3\n",
        "E5\n",
        "L180\n",
        "F1\n",
        "L90\n",
        "E4\n",
        "L90\n",
        "S4\n",
        "F58\n",
        "N2\n",
        "F95\n",
        "L180\n",
        "F27\n",
        "S4\n",
        "L90\n",
        "E4\n",
        "F90\n",
        "E5\n",
        "N2\n",
        "F36\n",
        "W2\n",
        "S5\n",
        "N3\n",
        "W4\n",
        "S1\n",
        "F58\n",
        "E4\n",
        "R90\n",
        "F60\n",
        "E1\n",
        "N2\n",
        "R90\n",
        "N1\n",
        "W2\n",
        "L90\n",
        "N3\n",
        "F16\n",
        "S2\n",
        "R90\n",
        "F3\n",
        "R270\n",
        "F98\n",
        "S4\n",
        "W1\n",
        "S4\n",
        "F42\n",
        "S5\n",
        "N2\n",
        "E1\n",
        "N4\n",
        "E3\n",
        "R180\n",
        "W3\n",
        "R180\n",
        "N4\n",
        "F47\n",
        "N5\n",
        "W5\n",
        "F55\n",
        "E2\n",
        "F12\n",
        "W3\n",
        "N1\n",
        "R90\n",
        "F55\n",
        "R180\n",
        "F89\n",
        "S4\n",
        "F38\n",
        "S5\n",
        "R180\n",
        "N2\n",
        "W4\n",
        "F16\n",
        "N2\n",
        "R180\n",
        "E2\n",
        "R180\n",
        "W1\n",
        "R90\n",
        "F50\n",
        "E1\n",
        "F36\n",
        "W4\n",
        "F79\n",
        "E3\n",
        "F19\n",
        "N4\n",
        "L270\n",
        "N5\n",
        "W3\n",
        "F48\n",
        "R90\n",
        "F36\n",
        "S5\n",
        "F100\n",
        "E3\n",
        "F55\n",
        "N3\n",
        "F16\n",
        "R180\n",
        "S5\n",
        "N3\n",
        "R180\n",
        "N1\n",
        "F8\n",
        "E5\n",
        "L90\n",
        "F21\n",
        "L90\n",
        "F80\n",
        "L90\n",
        "N5\n",
        "W5\n",
        "S5\n",
        "E5\n",
        "R90\n",
        "N4\n",
        "E1\n",
        "F20\n",
        "S3\n",
        "L90\n",
        "E1\n",
        "N4\n",
        "F16\n",
        "F26\n",
        "E1\n",
        "F75\n",
        "E1\n",
        "R90\n",
        "F69\n",
        "N1\n",
        "E3\n",
        "R90\n",
        "F62\n",
        "L180\n",
        "W4\n",
        "F80\n",
        "S5\n",
        "L90\n",
        "F90\n",
        "W5\n",
        "F40\n",
        "W1\n",
        "L90\n",
        "W1\n",
        "R180\n",
        "S2\n",
        "F65\n",
        "E2\n",
        "R180\n",
        "E1\n",
        "F37\n",
        "E5\n",
        "R90\n",
        "N3\n",
        "W4\n",
        "S5\n",
        "E3\n",
        "F36\n",
        "F8\n",
        "W5\n",
        "L90\n",
        "F54\n",
        "N5\n",
        "L180\n",
        "S4\n",
        "L90\n",
        "F88\n",
        "R180\n",
        "F46\n",
        "N5\n",
        "E3\n",
        "F93\n",
        "S2\n",
        "E5\n",
        "R270\n",
        "F15\n",
        "R90\n",
        "F28\n",
        "S2\n",
        "F61\n",
        "N5\n",
        "L90\n",
        "W3\n",
        "S5\n",
        "L90\n",
        "N1\n",
        "L90\n",
        "F41\n",
        "W4\n",
        "N3\n",
        "F4\n",
        "N3\n",
        "F35\n",
        "R90\n",
        "W5\n",
        "F69\n",
        "L90\n",
        "E3\n",
        "R90\n",
        "N2\n",
        "W4\n",
        "N5\n",
        "E4\n",
        "R90\n",
        "N1\n",
        "R90\n",
        "N2\n",
        "F44\n",
        "L90\n",
        "F15\n",
        "W2\n",
        "R180\n",
        "E3\n",
        "L90\n",
        "F26\n",
        "L90\n",
        "S4\n",
        "W4\n",
        "S1\n",
        "F46\n",
        "E2\n",
        "S5\n",
        "R180\n",
        "W3\n",
        "F6\n",
        "L90\n",
        "W3\n",
        "F99\n",
        "E3\n",
        "L90\n",
        "S1\n",
        "E1\n",
        "N2\n",
        "L90\n",
        "N3\n",
        "E4\n",
        "N3\n",
        "W1\n",
        "F87\n",
        "N2\n",
        "L180\n",
        "E1\n",
        "S1\n",
        "E1\n",
        "N4\n",
        "F20\n",
        "N1\n",
        "L90\n",
        "F41\n",
        "N1\n",
        "E5\n",
        "F20\n",
        "N4\n",
        "R180\n",
        "L180\n",
        "F69\n",
        "N3\n",
        "L90\n",
        "F80\n",
        "W4\n",
        "N5\n",
        "F95\n",
        "L90\n",
        "F27\n",
        "N2\n",
        "F22\n",
        "L270\n",
        "F74\n",
        "W2\n",
        "R180\n",
        "S3\n",
        "F64\n",
        "N3\n",
        "L270\n",
        "E2\n",
        "L90\n",
        "E2\n",
        "N1\n",
        "F72\n",
        "W3\n",
        "F78\n",
        "S3\n",
        "F56\n",
        "E3\n",
        "S4\n",
        "R180\n",
        "S1\n",
        "F53\n",
        "E4\n",
        "F96\n",
        "N1\n",
        "L90\n",
        "N3\n",
        "E5\n",
        "F83\n",
        "S1\n",
        "F26\n",
        "S3\n",
        "W5\n",
        "F20\n",
        "R270\n",
        "W1\n",
        "R90\n",
        "N4\n",
        "W1\n",
        "F69\n",
        "R90\n",
        "F7\n",
        "R180\n",
        "F15\n",
        "R90\n",
        "F65\n",
        "S1\n",
        "F68\n",
        "S5\n",
        "R180\n",
        "E5\n",
        "L90\n",
        "S1\n",
        "F68\n",
        "N1\n",
        "F7\n",
        "R90\n",
        "S1\n",
        "W5\n",
        "F29\n",
        "R270\n",
        "E5\n",
        "S4\n",
        "F14\n",
        "W2\n",
        "F91\n",
        "E2\n",
        "S5\n",
        "E4\n",
        "L90\n",
        "F81\n",
        "S4\n",
        "F69\n",
        "R90\n",
        "W3\n",
        "R180\n",
        "F33\n",
        "L180\n",
        "F52\n",
        "S2\n",
        "W5\n",
        "F2\n",
        "N4\n",
        "F13\n",
        "L270\n",
        "F63\n",
        "N5\n",
        "S3\n",
        "W5\n",
        "F67\n",
        "S1\n",
        "L90\n",
        "E5\n",
        "L90\n",
        "S3\n",
        "W5\n",
        "F67\n",
        "R90\n",
        "F64\n",
        "S4\n",
        "E5\n",
        "N4\n",
        "F97\n",
        "E2\n",
        "R90\n",
        "N1\n",
        "F5\n",
        "S3\n",
        "L90\n",
        "S2\n",
        "R90\n",
        "N3\n",
        "E3\n",
        "F76\n",
        "S4\n",
        "R90\n",
        "W1\n",
        "L180\n",
        "W4\n",
        "L90\n",
        "S3\n",
        "L90\n",
        "F66\n",
        "L90\n",
        "E1\n",
        "F71\n",
        "L270\n",
        "F42\n",
        "N1\n",
        "W1\n",
        "L90\n",
        "S4\n",
        "E3\n",
        "F41\n",
        "R180\n",
        "F68\n",
        "W5\n",
        "L90\n",
        "N5\n",
        "E5\n",
        "F96\n",
        "W3\n",
        "F90\n",
        "S1\n",
        "E1\n",
        "R90\n",
        "N3\n",
        "F38\n",
        "R90\n",
        "W3\n",
        "R90\n",
        "W1\n",
        "L90\n",
        "S5\n",
        "F76\n",
        "E3\n",
        "F36\n",
        "S1\n",
        "L180\n",
        "E1\n",
        "F22\n",
        "N1\n",
        "W1\n",
        "R270\n",
        "E4\n",
        "N5\n",
        "E3\n",
        "N3\n",
        "W2\n",
        "L180\n",
        "S2\n",
        "W5\n",
        "N4\n",
        "W5\n",
        "N3\n",
        "R90\n",
        "S2\n",
        "F45\n",
        "R90\n",
        "W2\n",
        "R90\n",
        "W3\n",
        "F75\n",
        "R180\n",
        "E4\n",
        "N1\n",
        "E4\n",
        "L90\n",
        "F59\n",
        "L90\n",
        "S2\n",
        "E2\n",
        "S2\n",
        "R90\n",
        "E2\n",
        "R90\n",
        "F37\n",
        "E2\n",
        "S1\n",
        "S5\n",
        "F97\n",
        "W4\n",
        "N1\n",
        "F58\n",
        "S2\n",
        "F14\n",
        "L90\n",
        "E3\n",
        "R90\n",
        "N1\n",
        "F4\n",
        "F28\n",
        "R180\n",
        "F55\n",
        "N2\n",
        "R90\n",
        "F48\n",
        "E4\n",
        "S4\n",
        "L90\n",
        "N4\n",
        "E1\n",
        "F67\n",
        "S4\n",
        "E5\n",
        "F16\n",
        "E5\n",
        "F82\n",
        "E4\n",
        "L270\n",
        "R90\n",
        "N5\n",
        "L90\n",
        "F77\n",
        "S3\n",
        "E1\n",
        "F85\n",
        "W3\n",
        "S1\n",
        "E2\n",
        "S5\n",
        "R90\n",
        "N2\n",
        "W5\n",
        "E1\n",
        "L90\n",
        "E3\n",
        "R90\n",
        "N5\n",
        "L270\n",
        "N5\n",
        "F77\n",
        "W4\n",
        "N5\n",
        "E4\n",
        "L90\n",
        "E2\n",
        "L180\n",
        "F13\n",
        "E5\n",
        "L270\n",
        "W1\n",
        "F52\n",
        "L90\n",
        "W3\n",
        "R90\n",
        "F23\n",
        "E1\n",
        "F25\n",
        "S2\n",
        "L180\n",
        "F80\n",
        "R180\n",
        "S5\n",
        "E3\n",
        "F90\n",
        "E4\n",
        "F60\n",
        "N4\n",
        "F75\n",
        "E3\n",
        "F10\n",
        "S3\n",
        "E1\n",
        "F8\n",
        "W5\n",
        "R90\n",
        "F20\n",
        "S3\n",
        "R90\n",
        "N2\n",
        "F41\n",
        "L90\n",
        "F8\n",
        "R270\n",
        "W1\n",
        "S1\n",
        "E2\n",
        "L180\n",
        "S4\n",
        "F52\n",
        "E5\n",
        "N2\n",
        "S3\n",
        "W1\n",
        "F67\n",
        "L90\n",
        "E3\n",
        "S2\n",
        "E2\n",
        "F30\n",
        "F12\n",
        "E3\n",
        "F11\n",
        "W4\n",
        "L90\n",
        "S4\n",
        "F5\n",
        "R90\n",
        "S5\n",
        "E2\n",
        "E3\n",
        "R90\n",
        "F37\n",
        "R90\n",
        "F64\n",
        "R90\n",
        "F68\n",
        "L90\n",
        "N1\n",
        "E2\n",
        "N4\n",
        "R180\n",
        "W4\n",
        "L90\n",
        "F68\n",
        "R90\n",
        "N5\n",
        "E2\n",
        "F54\n",
        "W2\n",
        "F84\n",
        "L90\n",
        "E3\n",
        "R90\n",
        "E2\n",
        "F29\n",
        "R90\n",
        "N1\n",
        "E2\n",
        "L90\n",
        "S5\n",
        "F90\n",
        "E2\n",
        "F11\n",
        "S3\n",
        "F99\n",
        "W3\n",
        "S5\n",
        "F94\n",
        "R90\n",
        "F36\n",
        "'''"
      ],
      "execution_count": 1,
      "outputs": []
    }
  ]
}