HoffmannChristian
diff --git a/‎2023/advent_of_code_2023.ipynb
Lines changed: 73 additions & 0 deletions b/‎2023/advent_of_code_2023.ipynb
Lines changed: 73 additions & 0 deletions
@@ -61,6 +61,79 @@
     "\n",
     "get_calibration_sum(lines)"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Day 2: Cube Conundrum\n",
+    "\n",
+    "What is the sum of the IDs of those games?"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def get_games(lines):\n",
+    "    games = []\n",
+    "    for game_str in [line.strip().split(':')[1].split(';') for line in lines]:\n",
+    "        subsets =[]\n",
+    "        for subset_str in game_str:\n",
+    "            subset = { 'blue': 0, 'green': 0, 'red': 0}\n",
+    "            for cube_str in subset_str.split(','):\n",
+    "                cubes = cube_str.split()\n",
+    "                subset[cubes[1]] = int(cubes[0])\n",
+    "            subsets.append(subset)\n",
+    "        games.append(subsets)\n",
+    "    return games\n",
+    "\n",
+    "def is_possible(subset):\n",
+    "    max_num_cubes = {'blue': 14, 'green': 13, 'red': 12}\n",
+    "    return all(subset[color] <= max_num_cubes[color] for color in max_num_cubes.keys())\n",
+    "\n",
+    "def get_possible_game_ids(games):\n",
+    "    possible_game_ids = []\n",
+    "    for game_id in range(len(games)):\n",
+    "        if all(is_possible(subset) for subset in games[game_id]):\n",
+    "            possible_game_ids.append(game_id+1)\n",
+    "    return possible_game_ids\n",
+    "\n",
+    "lines = open('02_input.txt', 'r').readlines()\n",
+    "\n",
+    "games = get_games(lines)\n",
+    "sum(get_possible_game_ids(games))"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "What is the sum of the power of these sets?"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from math import prod\n",
+    "\n",
+    "def get_powers(games):\n",
+    "    powers = []\n",
+    "    for game in games:\n",
+    "        min_cube_num = {'blue': 0, 'green': 0, 'red': 0}\n",
+    "        for subset in game:\n",
+    "            for color in subset:\n",
+    "                min_cube_num[color] = max(min_cube_num[color], subset[color])\n",
+    "        powers.append(prod(min_cube_num.values()))\n",
+    "    return powers\n",
+    "\n",
+    "sum(get_powers(games))"
+   ]
   }
  ],
  "metadata": {