Day 14 - Restroom Redoubt.ipynb

{
 "cells": [
  {
   "cell_type": "raw",
   "id": "c25e3db0-3d96-42da-a348-41d8d5f027f8",
   "metadata": {},
   "source": [
    "Day 14 I learned how to use complex numbers to store positions/velocities on a 2D grid."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "c9b4fcd7-9355-4911-a6fb-813b670dd4a9",
   "metadata": {},
   "outputs": [],
   "source": [
    "with open(\"day_14_input_1.txt\") as fname:\n",
    "    robots = fname.readlines()\n",
    "\n",
    "WIDTH = 101\n",
    "HEIGHT = 103\n",
    "#WIDTH = 11\n",
    "#HEIGHT = 7\n",
    "\n",
    "# input looks like\n",
    "# p=0,4 v=3,-3\n",
    "# p=6,3 v=-1,-3"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "1bcd11e1-6a79-4f0e-bd2d-c3903db84938",
   "metadata": {},
   "outputs": [],
   "source": [
    "# parse input into two lists\n",
    "\n",
    "positions = []\n",
    "velocities = []\n",
    "for line in robots:\n",
    "    splitline = line.strip().split(\" v=\")\n",
    "\n",
    "    # get the comma separated numbers from the first part of the line\n",
    "    position = splitline[0].replace(\"p=\",\"\").split(\",\")\n",
    "\n",
    "    # convert to a complex number and add to the list\n",
    "    positions.append(complex(float(position[0]), float(position[1])))\n",
    "    \n",
    "    velocity = splitline[1].split(\",\")\n",
    "    velocities.append(complex(float(velocity[0]), float(velocity[1])))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "a5eca2ae-dc23-4925-ad2c-75182078185e",
   "metadata": {},
   "outputs": [],
   "source": [
    "# make sure we wrap around the position the grid\n",
    "\n",
    "def wrap(p):\n",
    "    if p.real < 0:\n",
    "        p = p + WIDTH\n",
    "    elif p.real >= WIDTH:\n",
    "        p = p - WIDTH\n",
    "    \n",
    "    if p.imag < 0:\n",
    "        p = p + HEIGHT*1j\n",
    "    elif p.imag >= HEIGHT:\n",
    "        p = p - HEIGHT*1j\n",
    "        \n",
    "    return p"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "0f348a46-3873-4728-ba87-63dd354a7e46",
   "metadata": {},
   "outputs": [],
   "source": [
    "# step through 100 iterations and return new position\n",
    "\n",
    "def walk_100(pos, vel):\n",
    "    for i in range(100):\n",
    "        pos += vel\n",
    "        pos = wrap(pos)\n",
    "    return pos"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "870f28cf-43cf-440a-aa1b-39e68bed1628",
   "metadata": {},
   "outputs": [],
   "source": [
    "finalpositions = []\n",
    "for p, v in zip(positions, velocities):\n",
    "    finalpositions.append(walk_100(p, v))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "c3398062-1dd2-411a-959f-bbef91260073",
   "metadata": {},
   "outputs": [],
   "source": [
    "# map STARTING positions into a dictionary grid\n",
    "\n",
    "dictionary = {}\n",
    "\n",
    "for pos in positions:\n",
    "    if dictionary.get(pos) is None: # blank space\n",
    "        dictionary[pos] = 1\n",
    "    else: # robot already there\n",
    "        dictionary[pos] += 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0e745e25-a3fe-4216-93db-b586f2ab33ea",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "# View the positions\n",
    "\n",
    "for y in range(HEIGHT):\n",
    "    string = \"\"\n",
    "    for x in range(WIDTH):\n",
    "        string = string + str(dictionary.get(x+y*1j, \".\"))\n",
    "    print(string)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "52ecf466-eca5-4aab-9b92-aabc57aabf4b",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "# map FINAL positions into a dictionary grid\n",
    "\n",
    "dictionary = {}\n",
    "\n",
    "for pos in finalpositions:\n",
    "    if dictionary.get(pos) is None: # blank space\n",
    "        dictionary[pos] = 1\n",
    "    else: # robot already there\n",
    "        dictionary[pos] += 1\n",
    "        \n",
    "# View the positions\n",
    "\n",
    "for y in range(HEIGHT):\n",
    "    string = \"\"\n",
    "    for x in range(WIDTH):\n",
    "        string = string + str(dictionary.get(x+y*1j, \".\"))\n",
    "    print(string)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "c9881b61-a7be-46bc-8987-0125dd30c8e0",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "117 123 127 125\n",
      "228457125\n"
     ]
    }
   ],
   "source": [
    "# Top left\n",
    "q1 = 0\n",
    "for y in range(HEIGHT//2):\n",
    "    for x in range(WIDTH//2):\n",
    "        q1 += dictionary.get(x+y*1j, 0)\n",
    "\n",
    "# Top right\n",
    "q2 = 0\n",
    "for y in range(HEIGHT//2):\n",
    "    for x in range(1+WIDTH//2, WIDTH):\n",
    "        q2 += dictionary.get(x+y*1j, 0)\n",
    "\n",
    "# Bottom left\n",
    "q3 = 0\n",
    "for y in range(1+HEIGHT//2, HEIGHT):\n",
    "    for x in range(WIDTH//2):\n",
    "        q3 += dictionary.get(x+y*1j, 0)\n",
    "\n",
    "# Bottom right\n",
    "q4 = 0\n",
    "for y in range(1+HEIGHT//2, HEIGHT):\n",
    "    for x in range(1+WIDTH//2, WIDTH):\n",
    "        q4 += dictionary.get(x+y*1j, 0)\n",
    "\n",
    "print(q1, q2, q3, q4)\n",
    "print(q1*q2*q3*q4)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "052b2272-8f62-4f0b-8b25-c852d4ae8677",
   "metadata": {},
   "source": [
    "## Part 2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "a83ff2b0-3eaf-46a4-97c2-340d7a149278",
   "metadata": {},
   "outputs": [],
   "source": [
    "def walk_1(pos, vel):\n",
    "    pos += vel\n",
    "    pos = wrap(pos)\n",
    "    return pos"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "f07734fb-8379-476e-982c-0e2a5282bc8e",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "SECONDS: 6493\n",
      "WEIGHT: 0.6869244935543278\n",
      ".....................................................................................................\n",
      "..................1..................................................................................\n",
      "..................1..................................................................................\n",
      "..............................................................1..1...........1.......................\n",
      ".................................................................1...................................\n",
      "................................................................1......................1.............\n",
      "................................................................1...1................................\n",
      "..........1................................................1.........................................\n",
      "1...1............................1..............................1.......................1............\n",
      "........................................1...................1...............................1....1.1.\n",
      "..........................1...........................................1..............................\n",
      ".....................................................................................................\n",
      ".......................1.............................................................................\n",
      ".....................................................................................................\n",
      "......................................................1..............................................\n",
      "............................................................................1.........1..............\n",
      "..1..................1...............................................................................\n",
      "..........................................................1..........................................\n",
      ".......................................................................................1........1....\n",
      ".................1...................................................................................\n",
      "........................................................1...........................1................\n",
      "...1................1............1...................................................................\n",
      "....................1........................................1......1................................\n",
      "......1.........1.................1...................................11.............................\n",
      "........................1...1....................................................1...................\n",
      ".....................................................................................................\n",
      "...............................1..............................................1......................\n",
      "..............................................1......................................................\n",
      ".....................................................................................................\n",
      ".............1..............1...........................................1............................\n",
      "......................................................1...................1..........................\n",
      "............................................................1............1...........................\n",
      "1............1..................................................................1....................\n",
      "..............1......................................................................................\n",
      ".....................................................................................................\n",
      ".............................1.......................................................................\n",
      "........................................1111111111111111111111111111111..............................\n",
      "........................................1.............................1..............................\n",
      "........................................1.............................1..............................\n",
      "........................................1.............................1..............................\n",
      "........................................1.............................1..............1...............\n",
      "...................1....................1..............1..............1..............................\n",
      "...1...1.......1........................1.............111.............1..............................\n",
      ".............1..........................1............11111............1..............................\n",
      "........................................1...........1111111...........1..............................\n",
      "........................................1..........111111111..........1...............1..............\n",
      "........................................1............11111............1..............................\n",
      "........................................1...........1111111...........1..................1...........\n",
      "..............1.........................1..........111111111..........1..............................\n",
      "...................................1....1.........11111111111.........1..............................\n",
      "........................................1........1111111111111........1..........1......1............\n",
      "..........................1.............1..........111111111..........1..............................\n",
      "........................................1.........11111111111.........1.1............................\n",
      "........................................1........1111111111111........1..............................\n",
      "........................................1.......111111111111111.......1..............................\n",
      "..............................1.1.......1......11111111111111111......1..............................\n",
      "........................................1........1111111111111........1..............................\n",
      "........................................1.......111111111111111.......1...........1..................\n",
      "......1.................................1......11111111111111111......1...1........1.................\n",
      "1.......................................1.....1111111111111111111.....1..............................\n",
      "........................................1....111111111111111111111....1...............1..............\n",
      ".11.....................................1.............111.............1..............................\n",
      "........................................1.............111.............1.......1......................\n",
      "........................................1.............111.............1..........1............1......\n",
      "........................................1.............................1..............................\n",
      "........................................1.............................1..............................\n",
      "........................................1.............................1..............................\n",
      "........................................1.............................1..............................\n",
      "........................................1111111111111111111111111111111..............................\n",
      ".....................................................................................................\n",
      ".......................................................1.............................................\n",
      "....1................1.................................1.............................1...............\n",
      "..........1.........................................................................................1\n",
      "...................................................................1.................................\n",
      "........................1.................................1..................1.......................\n",
      "......................................................1................................1.............\n",
      ".....................................................................................................\n",
      "11...........................................................................................1.......\n",
      "..........................................................1..........................................\n",
      ".......................1.............................................................1...............\n",
      ".....................................................................................................\n",
      "..................................1...........................1......................................\n",
      "....................................................................1................................\n",
      "................1.....................................................................1..............\n",
      "...............................................................................1.....................\n",
      ".........1.........1.....................................1...............1...........................\n",
      "......................................1..............................................................\n",
      "...............................1........1........................................1...................\n",
      "..............................................1......1........................................1......\n",
      "................................................................................................1....\n",
      "............................................................................................1........\n",
      ".....1...............................................................................................\n",
      ".................1...................................................................................\n",
      ".1...........................................................1....................1....1.......1.....\n",
      "............................................................................................1.....1..\n",
      "......................................................................1.......................1......\n",
      "......................................1..............................................................\n",
      "...................................................................................1.................\n",
      ".....................................................................................................\n",
      ".....................................................................................................\n",
      ".....................................................................................................\n",
      ".............1.....................1.....1...........................................................\n",
      "..................................................................................1..................\n"
     ]
    }
   ],
   "source": [
    "for i in range(0,10000):\n",
    "    finalpositions = []\n",
    "    for p, v in zip(positions, velocities):\n",
    "        finalpositions.append(walk_1(p, v))\n",
    "    \n",
    "    dictionary = {}\n",
    "    \n",
    "    for pos in finalpositions:\n",
    "        if dictionary.get(pos) is None: # blank space\n",
    "            dictionary[pos] = 1\n",
    "        else: # robot already there\n",
    "            dictionary[pos] += 1\n",
    "\n",
    "    # Top\n",
    "    q1 = 0\n",
    "    for y in range(HEIGHT//4):\n",
    "        for x in range(WIDTH):\n",
    "            q1 += dictionary.get(x+y*1j, 0)\n",
    "    \n",
    "    # Bottom\n",
    "    q2 = 0\n",
    "    for y in range(3*HEIGHT//4):\n",
    "        for x in range(WIDTH):\n",
    "            q2 += dictionary.get(x+y*1j, 0)\n",
    "    \n",
    "    # Left side\n",
    "    q3 = 0\n",
    "    for y in range(HEIGHT//4, 3*HEIGHT//4):\n",
    "        for x in range(WIDTH//4):\n",
    "            q3 += dictionary.get(x+y*1j, 0)\n",
    "    \n",
    "    # Right side\n",
    "    q4 = 0\n",
    "    for y in range(HEIGHT//4, 3*HEIGHT//4):\n",
    "        for x in range(3*WIDTH//4, WIDTH):\n",
    "            q4 += dictionary.get(x+y*1j, 0)\n",
    "\n",
    "    # Middle\n",
    "    mid = 0\n",
    "    for y in range(HEIGHT//4, 3*HEIGHT//4):\n",
    "        for x in range(WIDTH//4, 3*WIDTH//4):\n",
    "            mid += dictionary.get(x+y*1j, 0)\n",
    "    \n",
    "    centre_mass = mid/(q1+q2+q3+q4)\n",
    "\n",
    "    if centre_mass > 0.5:\n",
    "        print(\"SECONDS:\",i+1)\n",
    "        print(\"WEIGHT:\",centre_mass)\n",
    "        for y in range(HEIGHT):\n",
    "            string = \"\"\n",
    "            for x in range(WIDTH):\n",
    "                string = string + str(dictionary.get(x+y*1j, \".\"))\n",
    "            print(string)\n",
    "        break\n",
    "\n",
    "    positions = finalpositions"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}