day_22.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "day-22.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "authorship_tag": "ABX9TyN5wWKRAAOS5HGWcSpQo/HO",
      "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_22.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": "IiPjJI272kYe",
        "outputId": "21b2dfd3-72fc-4df6-dd0d-a5422552ed92"
      },
      "source": [
        "# part 1\n",
        "p1 = []\n",
        "p2 = []\n",
        "\n",
        "# parse\n",
        "p = p1\n",
        "for line in input_string.splitlines():\n",
        "  if line.startswith(\"Player\"):\n",
        "    continue\n",
        "  if not line:\n",
        "    p = p2\n",
        "    continue\n",
        "  p.append(int(line))\n",
        "\n",
        "while len(p1) > 0 and len(p2) > 0:\n",
        "  # draw cards\n",
        "  p1c = p1.pop(0)\n",
        "  p2c = p2.pop(0)\n",
        "\n",
        "  # compare cards\n",
        "  if p1c > p2c:\n",
        "    # put on bottom, winning card first\n",
        "    p1.append(p1c)\n",
        "    p1.append(p2c)\n",
        "  if p1c < p2c:\n",
        "    p2.append(p2c)\n",
        "    p2.append(p1c)\n",
        "\n",
        "print('p1', p1)\n",
        "print('p2', p2)\n",
        "# Once the game ends, you can calculate the winning player's score. The bottom card in their deck is worth the value\n",
        "# of the card multiplied by 1, the second-from-the-bottom card is worth the value of the card multiplied by 2,\n",
        "# and so on.\n",
        "p = p1 if len(p1) > 0 else p2\n",
        "\n",
        "p.reverse()\n",
        "score = 0\n",
        "for i, c in enumerate(p, start=1):\n",
        "  score += c * i\n",
        "\n",
        "print('score:', score)\n"
      ],
      "execution_count": 2,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "p1 [48, 30, 42, 19, 45, 26, 28, 12, 41, 11, 32, 8, 9, 7, 46, 15, 31, 1, 35, 25, 44, 24, 33, 10, 38, 29, 23, 13, 27, 6, 4, 3, 18, 5, 16, 2, 50, 37, 47, 21, 43, 20, 49, 36, 39, 22, 40, 17, 34, 14]\n",
            "p2 []\n",
            "score: 32489\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "ZCaRNmwE2qVQ",
        "outputId": "067ac3fe-46e8-4a90-ca0f-c74f3cfdf91a"
      },
      "source": [
        "# from day_22_input import *\n",
        "import copy\n",
        "\n",
        "p1_deck = []\n",
        "p2_deck = []\n",
        "\n",
        "test2_string = '''Player 1:\n",
        "43\n",
        "19\n",
        "\n",
        "Player 2:\n",
        "2\n",
        "29\n",
        "14\n",
        "'''\n",
        "# parse\n",
        "p = p1_deck\n",
        "# for line in test2_string.splitlines():\n",
        "for line in input_string.splitlines():\n",
        "  if line.startswith(\"Player\"):\n",
        "    continue\n",
        "  if not line:\n",
        "    p = p2_deck\n",
        "    continue\n",
        "  p.append(int(line))\n",
        "\n",
        "\n",
        "def play_game(p1, p2, depth):\n",
        "  played_games = set()\n",
        "  while True:\n",
        "\n",
        "    # Check if someone has won all the cards\n",
        "    if not p1:\n",
        "      return 2\n",
        "    elif not p2:\n",
        "      return 1\n",
        "\n",
        "    # break infinite loops\n",
        "    new_deck_state = hash(tuple(p1 +[0]+ p2))\n",
        "    if new_deck_state in played_games:\n",
        "      winner = 1\n",
        "      # draw cards\n",
        "      p1c = p1.pop(0)\n",
        "      p2c = p2.pop(0)\n",
        "    else:\n",
        "      played_games.add(hash(tuple(p1 +[0]+ p2)))\n",
        "      if depth == 0:\n",
        "        if len(played_games)%10 == 0:\n",
        "          print(f'playing game {len(played_games)} at depth {depth}' )\n",
        "\n",
        "      # draw cards\n",
        "      p1c = p1.pop(0)\n",
        "      p2c = p2.pop(0)\n",
        "\n",
        "      # recursive game\n",
        "      if len(p1) >= p1c and len(p2) >= p2c:\n",
        "        # if there was a previous round in this game that had exactly the same cards in the same order in the same\n",
        "        # players' decks, the game instantly ends in a win for player 1.\n",
        "        winner = play_game(copy.deepcopy(p1[:p1c]), copy.deepcopy(p2[:p2c]), depth=depth+1)\n",
        "\n",
        "      # regular game\n",
        "      else:\n",
        "        winner = 1 if p1c > p2c else 2 # compare cards\n",
        "\n",
        "    if winner == 1: p1.extend([p1c, p2c]) # put on bottom, winning card first\n",
        "    if winner == 2: p2.extend([p2c, p1c])\n",
        "\n",
        "  return winner\n",
        "\n",
        "play_game(p1_deck, p2_deck, 0)\n",
        "\n",
        "\n",
        "print('p1_deck', p1_deck)\n",
        "print('p2_deck', p2_deck)\n",
        "# Once the game ends, you can calculate the winning player's score. The bottom card in their deck is worth the value\n",
        "# of the card multiplied by 1, the second-from-the-bottom card is worth the value of the card multiplied by 2,\n",
        "# and so on.\n",
        "p = p1_deck if len(p1_deck) > 0 else p2_deck\n",
        "\n",
        "p.reverse()\n",
        "score = 0\n",
        "for i, c in enumerate(p, start=1):\n",
        "  score += c * i\n",
        "\n",
        "print('score:', score)\n"
      ],
      "execution_count": 6,
      "outputs": [
        {
          "output_type": "stream",
          "text": [
            "playing game 10 at depth 0\n",
            "playing game 20 at depth 0\n",
            "playing game 30 at depth 0\n",
            "playing game 40 at depth 0\n",
            "playing game 50 at depth 0\n",
            "playing game 60 at depth 0\n",
            "playing game 70 at depth 0\n",
            "playing game 80 at depth 0\n",
            "playing game 90 at depth 0\n",
            "playing game 100 at depth 0\n",
            "playing game 110 at depth 0\n",
            "playing game 120 at depth 0\n",
            "playing game 130 at depth 0\n",
            "playing game 140 at depth 0\n",
            "playing game 150 at depth 0\n",
            "playing game 160 at depth 0\n",
            "playing game 170 at depth 0\n",
            "playing game 180 at depth 0\n",
            "playing game 190 at depth 0\n",
            "playing game 200 at depth 0\n",
            "playing game 210 at depth 0\n",
            "playing game 220 at depth 0\n",
            "playing game 230 at depth 0\n",
            "playing game 240 at depth 0\n",
            "playing game 250 at depth 0\n",
            "playing game 260 at depth 0\n",
            "playing game 270 at depth 0\n",
            "playing game 280 at depth 0\n",
            "playing game 290 at depth 0\n",
            "playing game 300 at depth 0\n",
            "playing game 310 at depth 0\n",
            "playing game 320 at depth 0\n",
            "playing game 330 at depth 0\n",
            "playing game 340 at depth 0\n",
            "playing game 350 at depth 0\n",
            "playing game 360 at depth 0\n",
            "playing game 370 at depth 0\n",
            "playing game 380 at depth 0\n",
            "playing game 390 at depth 0\n",
            "playing game 400 at depth 0\n",
            "playing game 410 at depth 0\n",
            "playing game 420 at depth 0\n",
            "playing game 430 at depth 0\n",
            "playing game 440 at depth 0\n",
            "playing game 450 at depth 0\n",
            "playing game 460 at depth 0\n",
            "playing game 470 at depth 0\n",
            "playing game 480 at depth 0\n",
            "playing game 490 at depth 0\n",
            "playing game 500 at depth 0\n",
            "playing game 510 at depth 0\n",
            "playing game 520 at depth 0\n",
            "playing game 530 at depth 0\n",
            "playing game 540 at depth 0\n",
            "playing game 550 at depth 0\n",
            "playing game 560 at depth 0\n",
            "playing game 570 at depth 0\n",
            "playing game 580 at depth 0\n",
            "playing game 590 at depth 0\n",
            "playing game 600 at depth 0\n",
            "playing game 610 at depth 0\n",
            "playing game 620 at depth 0\n",
            "playing game 630 at depth 0\n",
            "playing game 640 at depth 0\n",
            "playing game 650 at depth 0\n",
            "playing game 660 at depth 0\n",
            "playing game 670 at depth 0\n",
            "playing game 680 at depth 0\n",
            "playing game 690 at depth 0\n",
            "playing game 700 at depth 0\n",
            "playing game 710 at depth 0\n",
            "playing game 720 at depth 0\n",
            "playing game 730 at depth 0\n",
            "playing game 740 at depth 0\n",
            "playing game 750 at depth 0\n",
            "p1_deck [38, 12, 45, 43, 49, 30, 39, 17, 48, 32, 46, 37, 23, 14, 41, 20, 11, 7, 42, 18, 29, 3, 40, 31, 16, 8, 13, 5, 47, 28, 33, 22, 26, 24, 15, 10, 6, 1, 44, 27, 34, 25, 36, 21, 19, 2, 50, 9, 35, 4]\n",
            "p2_deck []\n",
            "score: 35676\n"
          ],
          "name": "stdout"
        }
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "oq_RLyIC2nJZ"
      },
      "source": [
        "test_string = '''Player 1:\n",
        "9\n",
        "2\n",
        "6\n",
        "3\n",
        "1\n",
        "\n",
        "Player 2:\n",
        "5\n",
        "8\n",
        "4\n",
        "7\n",
        "10\n",
        "'''\n",
        "\n",
        "input_string = '''Player 1:\n",
        "41\n",
        "26\n",
        "29\n",
        "11\n",
        "50\n",
        "38\n",
        "42\n",
        "20\n",
        "13\n",
        "9\n",
        "40\n",
        "43\n",
        "10\n",
        "24\n",
        "35\n",
        "30\n",
        "23\n",
        "15\n",
        "31\n",
        "48\n",
        "27\n",
        "44\n",
        "16\n",
        "12\n",
        "14\n",
        "\n",
        "Player 2:\n",
        "18\n",
        "6\n",
        "32\n",
        "37\n",
        "25\n",
        "21\n",
        "33\n",
        "28\n",
        "7\n",
        "8\n",
        "45\n",
        "46\n",
        "49\n",
        "5\n",
        "19\n",
        "2\n",
        "39\n",
        "4\n",
        "17\n",
        "3\n",
        "22\n",
        "1\n",
        "34\n",
        "36\n",
        "47\n",
        "'''"
      ],
      "execution_count": 1,
      "outputs": []
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "WK3ldHczK6H8"
      },
      "source": [
        ""
      ],
      "execution_count": null,
      "outputs": []
    }
  ]
}