2022/14/day.14.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "5995cf6f-8554-4c0f-b3e4-abc149dc6400",
   "metadata": {},
   "source": [
    "# Day 14: Regolith Reservoir\n",
    "\n",
    "[https://adventofcode.com/2022/day/14](https://adventofcode.com/2022/day/14)\n",
    "\n",
    "## Description\n",
    "\n",
    "### Part One\n",
    "\n",
    "The distress signal leads you to a giant waterfall! Actually, hang on - the signal seems like it's coming from the waterfall itself, and that doesn't make any sense. However, you do notice a little path that leads _behind_ the waterfall.\n",
    "\n",
    "Correction: the distress signal leads you behind a giant waterfall! There seems to be a large cave system here, and the signal definitely leads further inside.\n",
    "\n",
    "As you begin to make your way deeper underground, you feel the ground rumble for a moment. Sand begins pouring into the cave! If you don't quickly figure out where the sand is going, you could quickly become trapped!\n",
    "\n",
    "Fortunately, your [familiarity](https://adventofcode.com/2018/day/17) with analyzing the path of falling material will come in handy here. You scan a two-dimensional vertical slice of the cave above you (your puzzle input) and discover that it is mostly _air_ with structures made of _rock_.\n",
    "\n",
    "Your scan traces the path of each solid rock structure and reports the `x,y` coordinates that form the shape of the path, where `x` represents distance to the right and `y` represents distance down. Each path appears as a single line of text in your scan. After the first point of each path, each point indicates the end of a straight horizontal or vertical line to be drawn from the previous point. For example:\n",
    "\n",
    "    498,4 -> 498,6 -> 496,6\n",
    "    503,4 -> 502,4 -> 502,9 -> 494,9\n",
    "    \n",
    "\n",
    "This scan means that there are two paths of rock; the first path consists of two straight lines, and the second path consists of three straight lines. (Specifically, the first path consists of a line of rock from `498,4` through `498,6` and another line of rock from `498,6` through `496,6`.)\n",
    "\n",
    "The sand is pouring into the cave from point `500,0`.\n",
    "\n",
    "Drawing rock as `#`, air as `.`, and the source of the sand as `+`, this becomes:\n",
    "\n",
    "    \n",
    "      4     5  5\n",
    "      9     0  0\n",
    "      4     0  3\n",
    "    0 ......+...\n",
    "    1 ..........\n",
    "    2 ..........\n",
    "    3 ..........\n",
    "    4 ....#...##\n",
    "    5 ....#...#.\n",
    "    6 ..###...#.\n",
    "    7 ........#.\n",
    "    8 ........#.\n",
    "    9 #########.\n",
    "    \n",
    "\n",
    "Sand is produced _one unit at a time_, and the next unit of sand is not produced until the previous unit of sand _comes to rest_. A unit of sand is large enough to fill one tile of air in your scan.\n",
    "\n",
    "A unit of sand always falls _down one step_ if possible. If the tile immediately below is blocked (by rock or sand), the unit of sand attempts to instead move diagonally _one step down and to the left_. If that tile is blocked, the unit of sand attempts to instead move diagonally _one step down and to the right_. Sand keeps moving as long as it is able to do so, at each step trying to move down, then down-left, then down-right. If all three possible destinations are blocked, the unit of sand _comes to rest_ and no longer moves, at which point the next unit of sand is created back at the source.\n",
    "\n",
    "So, drawing sand that has come to rest as `o`, the first unit of sand simply falls straight down and then stops:\n",
    "\n",
    "    ......+...\n",
    "    ..........\n",
    "    ..........\n",
    "    ..........\n",
    "    ....#...##\n",
    "    ....#...#.\n",
    "    ..###...#.\n",
    "    ........#.\n",
    "    ......o.#.\n",
    "    #########.\n",
    "    \n",
    "\n",
    "The second unit of sand then falls straight down, lands on the first one, and then comes to rest to its left:\n",
    "\n",
    "    ......+...\n",
    "    ..........\n",
    "    ..........\n",
    "    ..........\n",
    "    ....#...##\n",
    "    ....#...#.\n",
    "    ..###...#.\n",
    "    ........#.\n",
    "    .....oo.#.\n",
    "    #########.\n",
    "    \n",
    "\n",
    "After a total of five units of sand have come to rest, they form this pattern:\n",
    "\n",
    "    ......+...\n",
    "    ..........\n",
    "    ..........\n",
    "    ..........\n",
    "    ....#...##\n",
    "    ....#...#.\n",
    "    ..###...#.\n",
    "    ......o.#.\n",
    "    ....oooo#.\n",
    "    #########.\n",
    "    \n",
    "\n",
    "After a total of 22 units of sand:\n",
    "\n",
    "    ......+...\n",
    "    ..........\n",
    "    ......o...\n",
    "    .....ooo..\n",
    "    ....#ooo##\n",
    "    ....#ooo#.\n",
    "    ..###ooo#.\n",
    "    ....oooo#.\n",
    "    ...ooooo#.\n",
    "    #########.\n",
    "    \n",
    "\n",
    "Finally, only two more units of sand can possibly come to rest:\n",
    "\n",
    "    ......+...\n",
    "    ..........\n",
    "    ......o...\n",
    "    .....ooo..\n",
    "    ....#ooo##\n",
    "    ...o#ooo#.\n",
    "    ..###ooo#.\n",
    "    ....oooo#.\n",
    "    .o.ooooo#.\n",
    "    #########.\n",
    "    \n",
    "\n",
    "Once all _`24`_ units of sand shown above have come to rest, all further sand flows out the bottom, falling into the endless void. Just for fun, the path any new sand takes before falling forever is shown here with `~`:\n",
    "\n",
    "    .......+...\n",
    "    .......~...\n",
    "    ......~o...\n",
    "    .....~ooo..\n",
    "    ....~#ooo##\n",
    "    ...~o#ooo#.\n",
    "    ..~###ooo#.\n",
    "    ..~..oooo#.\n",
    "    .~o.ooooo#.\n",
    "    ~#########.\n",
    "    ~..........\n",
    "    ~..........\n",
    "    ~..........\n",
    "    \n",
    "\n",
    "Using your scan, simulate the falling sand. _How many units of sand come to rest before sand starts flowing into the abyss below?_"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "266555d2-b061-4d3f-b10e-565b1a32315c",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "696\n"
     ]
    }
   ],
   "source": [
    "from ast import literal_eval\n",
    "\n",
    "SOURCE_LOCATION = (500, 0)\n",
    "AIR = \".\"\n",
    "ROCK = \"#\"\n",
    "SAND = \"o\"\n",
    "SOURCE = \"+\"\n",
    "\n",
    "def neighbours(x, y):\n",
    "    for dx in [0, -1, 1]:\n",
    "        yield x + dx, y + 1\n",
    "\n",
    "def release_sand(cave, sand, condition, space_is_open):\n",
    "    while condition():\n",
    "        if open_neighbours := [n for n in neighbours(*sand[-1]) if space_is_open(n)]:\n",
    "            sand.append(open_neighbours[0])\n",
    "        else:\n",
    "            cave[sand.pop()] = SAND\n",
    "\n",
    "def is_inside(x, y):\n",
    "    return x_min <= x <= x_max and y_min <= y <= y_max            \n",
    "\n",
    "cave = open(\"day.14.input.txt\", \"r\").read().split('\\n')\n",
    "cave = [[literal_eval(coord) for coord in line.split(\" -> \")] for line in cave]\n",
    "cave = {\n",
    "    (x, y): ROCK\n",
    "    for path in cave\n",
    "    for (x1, y1), (x2, y2) in zip(path, path[1:])\n",
    "    for x in range(min(x1, x2), max(x1, x2) + 1)\n",
    "    for y in range(min(y1, y2), max(y1, y2) + 1)\n",
    "}\n",
    "cave[SOURCE_LOCATION] = SOURCE\n",
    "\n",
    "x_min, x_max = sorted(key[0] for key in cave)[::len(cave) - 1]\n",
    "y_min, y_max = sorted(key[1] for key in cave)[::len(cave) - 1]\n",
    "\n",
    "sand_path = [SOURCE_LOCATION]\n",
    "release_sand(cave, sand_path, lambda: is_inside(*sand_path[-1]), lambda r: r not in cave)\n",
    "print(sum(1 for val in cave.values() if val == SAND))"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "326847bb-63f5-4c69-84c6-bd2f836f8c3c",
   "metadata": {},
   "source": [
    "### Part Two\n",
    "\n",
    "You realize you misread the scan. There isn't an <span title=\"Endless Void is my C cover band.\">endless void</span> at the bottom of the scan - there's floor, and you're standing on it!\n",
    "\n",
    "You don't have time to scan the floor, so assume the floor is an infinite horizontal line with a `y` coordinate equal to _two plus the highest `y` coordinate_ of any point in your scan.\n",
    "\n",
    "In the example above, the highest `y` coordinate of any point is `9`, and so the floor is at `y=11`. (This is as if your scan contained one extra rock path like `-infinity,11 -> infinity,11`.) With the added floor, the example above now looks like this:\n",
    "\n",
    "            ...........+........\n",
    "            ....................\n",
    "            ....................\n",
    "            ....................\n",
    "            .........#...##.....\n",
    "            .........#...#......\n",
    "            .......###...#......\n",
    "            .............#......\n",
    "            .............#......\n",
    "            .....#########......\n",
    "            ....................\n",
    "    <-- etc #################### etc -->\n",
    "    \n",
    "\n",
    "To find somewhere safe to stand, you'll need to simulate falling sand until a unit of sand comes to rest at `500,0`, blocking the source entirely and stopping the flow of sand into the cave. In the example above, the situation finally looks like this after _`93`_ units of sand come to rest:\n",
    "\n",
    "    ............o............\n",
    "    ...........ooo...........\n",
    "    ..........ooooo..........\n",
    "    .........ooooooo.........\n",
    "    ........oo#ooo##o........\n",
    "    .......ooo#ooo#ooo.......\n",
    "    ......oo###ooo#oooo......\n",
    "    .....oooo.oooo#ooooo.....\n",
    "    ....oooooooooo#oooooo....\n",
    "    ...ooo#########ooooooo...\n",
    "    ..ooooo.......ooooooooo..\n",
    "    #########################\n",
    "    \n",
    "\n",
    "Using your scan, simulate the falling sand until the source of the sand becomes blocked. _How many units of sand come to rest?_\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "fe3976f0-562d-4681-9c34-b2933748f577",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "23610\n"
     ]
    }
   ],
   "source": [
    "floor = y_max + 2\n",
    "release_sand(cave, sand_path, lambda: sand_path, lambda r: r[1] < floor and r not in cave)\n",
    "print(sum(1 for val in cave.values() if val == SAND))"
   ]
  }
 ],
 "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.9.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}