day.22.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Day 22: Crab Combat\n",
    "\n",
    "It only takes a few hours of sailing the ocean on a raft for boredom to sink in. Fortunately, you brought a small deck of [space cards](https://adventofcode.com/2019/day/22)! You'd like to play a game of **Combat**, and there's even an opponent available: a small crab that climbed aboard your raft before you left.\n",
    "\n",
    "Fortunately, it doesn't take long to teach the crab the rules.\n",
    "\n",
    "Before the game starts, split the cards so each player has their own deck (your puzzle input). Then, the game consists of a series of **rounds**: both players draw their top card, and the player with the higher-valued card wins the round. The winner keeps both cards, placing them on the bottom of their own deck so that the winner's card is above the other card. If this causes a player to have all of the cards, they win, and the game ends.\n",
    "\n",
    "For example, consider the following starting decks:\n",
    "\n",
    "    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",
    "This arrangement means that player 1's deck contains 5 cards, with `9` on top and `1` on the bottom; player 2's deck also contains 5 cards, with `5` on top and `10` on the bottom.\n",
    "\n",
    "The first round begins with both players drawing the top card of their decks: `9` and `5`. Player 1 has the higher card, so both cards move to the bottom of player 1's deck such that `9` is above `5`. In total, it takes 29 rounds before a player has all of the cards:\n",
    "\n",
    "    -- Round 1 --\n",
    "    Player 1's deck: 9, 2, 6, 3, 1\n",
    "    Player 2's deck: 5, 8, 4, 7, 10\n",
    "    Player 1 plays: 9\n",
    "    Player 2 plays: 5\n",
    "    Player 1 wins the round!\n",
    "\n",
    "    -- Round 2 --\n",
    "    Player 1's deck: 2, 6, 3, 1, 9, 5\n",
    "    Player 2's deck: 8, 4, 7, 10\n",
    "    Player 1 plays: 2\n",
    "    Player 2 plays: 8\n",
    "    Player 2 wins the round!\n",
    "\n",
    "    -- Round 3 --\n",
    "    Player 1's deck: 6, 3, 1, 9, 5\n",
    "    Player 2's deck: 4, 7, 10, 8, 2\n",
    "    Player 1 plays: 6\n",
    "    Player 2 plays: 4\n",
    "    Player 1 wins the round!\n",
    "\n",
    "    -- Round 4 --\n",
    "    Player 1's deck: 3, 1, 9, 5, 6, 4\n",
    "    Player 2's deck: 7, 10, 8, 2\n",
    "    Player 1 plays: 3\n",
    "    Player 2 plays: 7\n",
    "    Player 2 wins the round!\n",
    "\n",
    "    -- Round 5 --\n",
    "    Player 1's deck: 1, 9, 5, 6, 4\n",
    "    Player 2's deck: 10, 8, 2, 7, 3\n",
    "    Player 1 plays: 1\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins the round!\n",
    "\n",
    "    ...several more rounds pass...\n",
    "\n",
    "    -- Round 27 --\n",
    "    Player 1's deck: 5, 4, 1\n",
    "    Player 2's deck: 8, 9, 7, 3, 2, 10, 6\n",
    "    Player 1 plays: 5\n",
    "    Player 2 plays: 8\n",
    "    Player 2 wins the round!\n",
    "\n",
    "    -- Round 28 --\n",
    "    Player 1's deck: 4, 1\n",
    "    Player 2's deck: 9, 7, 3, 2, 10, 6, 8, 5\n",
    "    Player 1 plays: 4\n",
    "    Player 2 plays: 9\n",
    "    Player 2 wins the round!\n",
    "\n",
    "    -- Round 29 --\n",
    "    Player 1's deck: 1\n",
    "    Player 2's deck: 7, 3, 2, 10, 6, 8, 5, 9, 4\n",
    "    Player 1 plays: 1\n",
    "    Player 2 plays: 7\n",
    "    Player 2 wins the round!\n",
    "\n",
    "\n",
    "    == Post-game results ==\n",
    "    Player 1's deck: \n",
    "    Player 2's deck: 3, 2, 10, 6, 8, 5, 9, 4, 7, 1\n",
    "\n",
    "Once the game ends, you can calculate the winning player's **score**. The bottom card in their deck is worth the value of the card multiplied by 1, the second-from-the-bottom card is worth the value of the card multiplied by 2, and so on. With 10 cards, the top card is worth the value on the card multiplied by 10. In this example, the winning player's score is:\n",
    "\n",
    "       3 * 10\n",
    "    +  2 *  9\n",
    "    + 10 *  8\n",
    "    +  6 *  7\n",
    "    +  8 *  6\n",
    "    +  5 *  5\n",
    "    +  9 *  4\n",
    "    +  4 *  3\n",
    "    +  7 *  2\n",
    "    +  1 *  1\n",
    "    = 306\n",
    "\n",
    "So, once the game ends, the winning player's score is **`306`**.\n",
    "\n",
    "Play the small crab in a game of Combat using the two decks you just dealt. **What is the winning player's score?**\n",
    "\n",
    "Your puzzle answer was `35397`."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "input = \"\"\"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\"\"\".split('\\n\\n')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "input = open('day.22.input.txt', 'r').read().split('\\n\\n')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "35397"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import numpy as np\n",
    "\n",
    "player1 = [int(s) for s in input[0].split('\\n')[1:]]\n",
    "player2 = [int(s) for s in input[1].split('\\n')[1:]]\n",
    "\n",
    "while (len(player1) > 0) & (len(player2) > 0):\n",
    "    c1 = player1[0]\n",
    "    c2 = player2[0]\n",
    "    player1 = player1[1:]\n",
    "    player2 = player2[1:]  \n",
    "    if c1 > c2:\n",
    "        player1 += [c1, c2]\n",
    "    else:\n",
    "        player2 += [c2, c1]\n",
    "\n",
    "winner = player1 + player2\n",
    "np.dot(winner, range(len(winner), 0, -1))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Part Two\n",
    "\n",
    "You lost to the small crab! Fortunately, crabs aren't very good at recursion. To defend your honor as a Raft Captain, you challenge the small crab to a game of **Recursive Combat**.\n",
    "\n",
    "Recursive Combat still starts by splitting the cards into two decks (you offer to play with the same starting decks as before - it's only fair). Then, the game consists of a series of **rounds with a few changes**:\n",
    "\n",
    "* Before either player deals a card, if there was a previous round in this game that had exactly the same cards in the same order in the same players' decks, the **game** instantly ends in a win for player 1. Previous rounds from other games are not considered. (This prevents infinite games of Recursive Combat, which everyone agrees is a bad idea.)\n",
    "* Otherwise, this round's cards must be in a new configuration; the players begin the round by each drawing the top card of their deck as normal.\n",
    "* If both players have at least as many cards remaining in their deck as the value of the card they just drew, the winner of the round is determined by playing a new game of Recursive Combat (see below).\n",
    "* Otherwise, at least one player must not have enough cards left in their deck to recurse; the winner of the round is the player with the higher-value card.\n",
    "\n",
    "As in regular Combat, the winner of the round (even if they won the round by winning a sub-game) takes the two cards dealt at the beginning of the round and places them on the bottom of their own deck (again so that the winner's card is above the other card). Note that the winner's card might be **the lower-valued of the two cards** if they won the round due to winning a sub-game. If collecting cards by winning the round causes a player to have all of the cards, they win, and the game ends.\n",
    "\n",
    "Here is an example of a small game that would loop forever without the infinite game prevention rule:\n",
    "\n",
    "    Player 1:\n",
    "    43\n",
    "    19\n",
    "\n",
    "    Player 2:\n",
    "    2\n",
    "    29\n",
    "    14\n",
    "\n",
    "During a round of Recursive Combat, if both players have at least as many cards in their own decks as the number on the card they just dealt, the winner of the round is determined by recursing into a sub-game of Recursive Combat. (For example, if player 1 draws the 3 card, and player 2 draws the 7 card, this would occur if player 1 has at least 3 cards left and player 2 has at least 7 cards left, not counting the 3 and 7 cards that were drawn.)\n",
    "\n",
    "To play a sub-game of Recursive Combat, each player creates a new deck by making a **copy** of the next cards in their deck (the quantity of cards copied is equal to the number on the card they drew to trigger the sub-game). During this sub-game, the game that triggered it is on hold and completely unaffected; no cards are removed from players' decks to form the sub-game. (For example, if player 1 drew the 3 card, their deck in the sub-game would be **copies** of the next three cards in their deck.)\n",
    "\n",
    "Here is a complete example of gameplay, where `Game 1` is the primary game of Recursive Combat:\n",
    "\n",
    "    === Game 1 ===\n",
    "\n",
    "    -- Round 1 (Game 1) --\n",
    "    Player 1's deck: 9, 2, 6, 3, 1\n",
    "    Player 2's deck: 5, 8, 4, 7, 10\n",
    "    Player 1 plays: 9\n",
    "    Player 2 plays: 5\n",
    "    Player 1 wins round 1 of game 1!\n",
    "\n",
    "    -- Round 2 (Game 1) --\n",
    "    Player 1's deck: 2, 6, 3, 1, 9, 5\n",
    "    Player 2's deck: 8, 4, 7, 10\n",
    "    Player 1 plays: 2\n",
    "    Player 2 plays: 8\n",
    "    Player 2 wins round 2 of game 1!\n",
    "\n",
    "    -- Round 3 (Game 1) --\n",
    "    Player 1's deck: 6, 3, 1, 9, 5\n",
    "    Player 2's deck: 4, 7, 10, 8, 2\n",
    "    Player 1 plays: 6\n",
    "    Player 2 plays: 4\n",
    "    Player 1 wins round 3 of game 1!\n",
    "\n",
    "    -- Round 4 (Game 1) --\n",
    "    Player 1's deck: 3, 1, 9, 5, 6, 4\n",
    "    Player 2's deck: 7, 10, 8, 2\n",
    "    Player 1 plays: 3\n",
    "    Player 2 plays: 7\n",
    "    Player 2 wins round 4 of game 1!\n",
    "\n",
    "    -- Round 5 (Game 1) --\n",
    "    Player 1's deck: 1, 9, 5, 6, 4\n",
    "    Player 2's deck: 10, 8, 2, 7, 3\n",
    "    Player 1 plays: 1\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 5 of game 1!\n",
    "\n",
    "    -- Round 6 (Game 1) --\n",
    "    Player 1's deck: 9, 5, 6, 4\n",
    "    Player 2's deck: 8, 2, 7, 3, 10, 1\n",
    "    Player 1 plays: 9\n",
    "    Player 2 plays: 8\n",
    "    Player 1 wins round 6 of game 1!\n",
    "\n",
    "    -- Round 7 (Game 1) --\n",
    "    Player 1's deck: 5, 6, 4, 9, 8\n",
    "    Player 2's deck: 2, 7, 3, 10, 1\n",
    "    Player 1 plays: 5\n",
    "    Player 2 plays: 2\n",
    "    Player 1 wins round 7 of game 1!\n",
    "\n",
    "    -- Round 8 (Game 1) --\n",
    "    Player 1's deck: 6, 4, 9, 8, 5, 2\n",
    "    Player 2's deck: 7, 3, 10, 1\n",
    "    Player 1 plays: 6\n",
    "    Player 2 plays: 7\n",
    "    Player 2 wins round 8 of game 1!\n",
    "\n",
    "    -- Round 9 (Game 1) --\n",
    "    Player 1's deck: 4, 9, 8, 5, 2\n",
    "    Player 2's deck: 3, 10, 1, 7, 6\n",
    "    Player 1 plays: 4\n",
    "    Player 2 plays: 3\n",
    "    Playing a sub-game to determine the winner...\n",
    "\n",
    "    === Game 2 ===\n",
    "\n",
    "    -- Round 1 (Game 2) --\n",
    "    Player 1's deck: 9, 8, 5, 2\n",
    "    Player 2's deck: 10, 1, 7\n",
    "    Player 1 plays: 9\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 1 of game 2!\n",
    "\n",
    "    -- Round 2 (Game 2) --\n",
    "    Player 1's deck: 8, 5, 2\n",
    "    Player 2's deck: 1, 7, 10, 9\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 1\n",
    "    Player 1 wins round 2 of game 2!\n",
    "\n",
    "    -- Round 3 (Game 2) --\n",
    "    Player 1's deck: 5, 2, 8, 1\n",
    "    Player 2's deck: 7, 10, 9\n",
    "    Player 1 plays: 5\n",
    "    Player 2 plays: 7\n",
    "    Player 2 wins round 3 of game 2!\n",
    "\n",
    "    -- Round 4 (Game 2) --\n",
    "    Player 1's deck: 2, 8, 1\n",
    "    Player 2's deck: 10, 9, 7, 5\n",
    "    Player 1 plays: 2\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 4 of game 2!\n",
    "\n",
    "    -- Round 5 (Game 2) --\n",
    "    Player 1's deck: 8, 1\n",
    "    Player 2's deck: 9, 7, 5, 10, 2\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 9\n",
    "    Player 2 wins round 5 of game 2!\n",
    "\n",
    "    -- Round 6 (Game 2) --\n",
    "    Player 1's deck: 1\n",
    "    Player 2's deck: 7, 5, 10, 2, 9, 8\n",
    "    Player 1 plays: 1\n",
    "    Player 2 plays: 7\n",
    "    Player 2 wins round 6 of game 2!\n",
    "    The winner of game 2 is player 2!\n",
    "\n",
    "    ...anyway, back to game 1.\n",
    "    Player 2 wins round 9 of game 1!\n",
    "\n",
    "    -- Round 10 (Game 1) --\n",
    "    Player 1's deck: 9, 8, 5, 2\n",
    "    Player 2's deck: 10, 1, 7, 6, 3, 4\n",
    "    Player 1 plays: 9\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 10 of game 1!\n",
    "\n",
    "    -- Round 11 (Game 1) --\n",
    "    Player 1's deck: 8, 5, 2\n",
    "    Player 2's deck: 1, 7, 6, 3, 4, 10, 9\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 1\n",
    "    Player 1 wins round 11 of game 1!\n",
    "\n",
    "    -- Round 12 (Game 1) --\n",
    "    Player 1's deck: 5, 2, 8, 1\n",
    "    Player 2's deck: 7, 6, 3, 4, 10, 9\n",
    "    Player 1 plays: 5\n",
    "    Player 2 plays: 7\n",
    "    Player 2 wins round 12 of game 1!\n",
    "\n",
    "    -- Round 13 (Game 1) --\n",
    "    Player 1's deck: 2, 8, 1\n",
    "    Player 2's deck: 6, 3, 4, 10, 9, 7, 5\n",
    "    Player 1 plays: 2\n",
    "    Player 2 plays: 6\n",
    "    Playing a sub-game to determine the winner...\n",
    "\n",
    "    === Game 3 ===\n",
    "\n",
    "    -- Round 1 (Game 3) --\n",
    "    Player 1's deck: 8, 1\n",
    "    Player 2's deck: 3, 4, 10, 9, 7, 5\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 3\n",
    "    Player 1 wins round 1 of game 3!\n",
    "\n",
    "    -- Round 2 (Game 3) --\n",
    "    Player 1's deck: 1, 8, 3\n",
    "    Player 2's deck: 4, 10, 9, 7, 5\n",
    "    Player 1 plays: 1\n",
    "    Player 2 plays: 4\n",
    "    Playing a sub-game to determine the winner...\n",
    "\n",
    "    === Game 4 ===\n",
    "\n",
    "    -- Round 1 (Game 4) --\n",
    "    Player 1's deck: 8\n",
    "    Player 2's deck: 10, 9, 7, 5\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 1 of game 4!\n",
    "    The winner of game 4 is player 2!\n",
    "\n",
    "    ...anyway, back to game 3.\n",
    "    Player 2 wins round 2 of game 3!\n",
    "\n",
    "    -- Round 3 (Game 3) --\n",
    "    Player 1's deck: 8, 3\n",
    "    Player 2's deck: 10, 9, 7, 5, 4, 1\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 3 of game 3!\n",
    "\n",
    "    -- Round 4 (Game 3) --\n",
    "    Player 1's deck: 3\n",
    "    Player 2's deck: 9, 7, 5, 4, 1, 10, 8\n",
    "    Player 1 plays: 3\n",
    "    Player 2 plays: 9\n",
    "    Player 2 wins round 4 of game 3!\n",
    "    The winner of game 3 is player 2!\n",
    "\n",
    "    ...anyway, back to game 1.\n",
    "    Player 2 wins round 13 of game 1!\n",
    "\n",
    "    -- Round 14 (Game 1) --\n",
    "    Player 1's deck: 8, 1\n",
    "    Player 2's deck: 3, 4, 10, 9, 7, 5, 6, 2\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 3\n",
    "    Player 1 wins round 14 of game 1!\n",
    "\n",
    "    -- Round 15 (Game 1) --\n",
    "    Player 1's deck: 1, 8, 3\n",
    "    Player 2's deck: 4, 10, 9, 7, 5, 6, 2\n",
    "    Player 1 plays: 1\n",
    "    Player 2 plays: 4\n",
    "    Playing a sub-game to determine the winner...\n",
    "\n",
    "    === Game 5 ===\n",
    "\n",
    "    -- Round 1 (Game 5) --\n",
    "    Player 1's deck: 8\n",
    "    Player 2's deck: 10, 9, 7, 5\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 1 of game 5!\n",
    "    The winner of game 5 is player 2!\n",
    "\n",
    "    ...anyway, back to game 1.\n",
    "    Player 2 wins round 15 of game 1!\n",
    "\n",
    "    -- Round 16 (Game 1) --\n",
    "    Player 1's deck: 8, 3\n",
    "    Player 2's deck: 10, 9, 7, 5, 6, 2, 4, 1\n",
    "    Player 1 plays: 8\n",
    "    Player 2 plays: 10\n",
    "    Player 2 wins round 16 of game 1!\n",
    "\n",
    "    -- Round 17 (Game 1) --\n",
    "    Player 1's deck: 3\n",
    "    Player 2's deck: 9, 7, 5, 6, 2, 4, 1, 10, 8\n",
    "    Player 1 plays: 3\n",
    "    Player 2 plays: 9\n",
    "    Player 2 wins round 17 of game 1!\n",
    "    The winner of game 1 is player 2!\n",
    "\n",
    "\n",
    "    == Post-game results ==\n",
    "    Player 1's deck: \n",
    "    Player 2's deck: 7, 5, 6, 2, 4, 1, 10, 8, 9, 3\n",
    "\n",
    "After the game, the winning player's score is calculated from the cards they have in their original deck using the same rules as regular Combat. In the above game, the winning player's score is **`291`**.\n",
    "\n",
    "Defend your honor as Raft Captain by playing the small crab in a game of Recursive Combat using the same two decks as before. **What is the winning player's score?**\n",
    "\n",
    "Your puzzle answer was `31120`."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "31120"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "player1 = [int(s) for s in input[0].split('\\n')[1:]]\n",
    "player2 = [int(s) for s in input[1].split('\\n')[1:]]\n",
    "\n",
    "def play_game(player1, player2):\n",
    "    known = set()\n",
    "    while (len(player1) > 0) & (len(player2) > 0):\n",
    "        k = tuple(player1 + [-1] + player2)\n",
    "        if k in known:\n",
    "            return [-1], []\n",
    "        else:\n",
    "            known.add(k)\n",
    "        c1 = player1[0]\n",
    "        c2 = player2[0]\n",
    "        player1 = player1[1:]\n",
    "        player2 = player2[1:]\n",
    "        if (len(player1) >= c1) & (len(player2) >= c2):\n",
    "            sub1, sub2 = play_game(player1[:c1], player2[:c2])\n",
    "            if len(sub1) > len(sub2):\n",
    "                player1 += [c1, c2]\n",
    "            else:\n",
    "                player2 += [c2, c1]\n",
    "            \n",
    "        else:\n",
    "            if c1 > c2:\n",
    "                player1 += [c1, c2]\n",
    "            else:\n",
    "                player2 += [c2, c1]\n",
    "    return player1, player2\n",
    "\n",
    "player1, player2 = play_game(player1, player2)\n",
    "\n",
    "winner = player1 + player2\n",
    "np.dot(winner, range(len(winner), 0, -1))"
   ]
  }
 ],
 "metadata": {
  "jupytext": {
   "hide_notebook_metadata": true
  },
  "kernelspec": {
   "display_name": "Python 3",
   "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.7.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}