{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 133,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "import collections\n",
    "import itertools"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 134,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def player_wins_fight(player, boss, verbose=False):\n",
    "    boss_hp = boss['hp']\n",
    "    pc_hp = player['hp']\n",
    "    while pc_hp > 0 and boss_hp > 0:\n",
    "        boss_hp -= max(player['damage'] - boss['armour'], 1)\n",
    "        pc_hp -= max(boss['damage'] - player['armour'], 1)\n",
    "        if verbose:\n",
    "            print('PC:', player['hp'], ' . Boss:', boss_hp)\n",
    "    if boss_hp <= 0:\n",
    "        return True\n",
    "    else:\n",
    "        return False"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 135,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "execution_count": 135,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pc = {'hp': 8, 'damage': 5, 'armour': 5}\n",
    "boss =  {'hp': 12, 'damage': 7, 'armour': 2}\n",
    "player_wins_fight(pc, boss)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 136,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "({'armour': 5, 'damage': 5, 'hp': 8}, {'armour': 2, 'damage': 7, 'hp': 12})"
      ]
     },
     "execution_count": 136,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pc, boss"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 137,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'armour': 2, 'damage': 8, 'hp': 100}"
      ]
     },
     "execution_count": 137,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "boss = {}\n",
    "for l in open('advent21.txt').readlines():\n",
    "    t, v = l.strip().split(': ')\n",
    "    if t == 'Hit Points': \n",
    "        boss['hp'] = int(v)\n",
    "    elif t == 'Armor': \n",
    "        boss['armour'] = int(v)\n",
    "    else:\n",
    "        boss[t.lower()] = int(v)\n",
    "boss"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 138,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "weapons_s = \"\"\"Dagger        8     4       0\n",
    "Shortsword   10     5       0\n",
    "Warhammer    25     6       0\n",
    "Longsword    40     7       0\n",
    "Greataxe     74     8       0\"\"\"\n",
    "\n",
    "armour_s = \"\"\"Leather      13     0       1\n",
    "Chainmail    31     0       2\n",
    "Splintmail   53     0       3\n",
    "Bandedmail   75     0       4\n",
    "Platemail   102     0       5\"\"\"\n",
    "\n",
    "rings_s = \"\"\"Damage+1    25     1       0\n",
    "Damage+2    50     2       0\n",
    "Damage+3   100     3       0\n",
    "Defense+1   20     0       1\n",
    "Defense+2   40     0       2\n",
    "Defense+3   80     0       3\"\"\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 140,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "def parse_table(table):\n",
    "    d = []\n",
    "    for l in table.split('\\n'):\n",
    "        ls = l.split()\n",
    "        d += [{'name': ls[0].strip(),\n",
    "               'cost': int(ls[1].strip()), \n",
    "               'damage': int(ls[2].strip()), \n",
    "               'armour': int(ls[3].strip()) }]\n",
    "    return d"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 141,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "([{'armour': 0, 'cost': 8, 'damage': 4, 'name': 'Dagger'},\n",
       "  {'armour': 0, 'cost': 10, 'damage': 5, 'name': 'Shortsword'},\n",
       "  {'armour': 0, 'cost': 25, 'damage': 6, 'name': 'Warhammer'},\n",
       "  {'armour': 0, 'cost': 40, 'damage': 7, 'name': 'Longsword'},\n",
       "  {'armour': 0, 'cost': 74, 'damage': 8, 'name': 'Greataxe'}],\n",
       " [{'armour': 1, 'cost': 13, 'damage': 0, 'name': 'Leather'},\n",
       "  {'armour': 2, 'cost': 31, 'damage': 0, 'name': 'Chainmail'},\n",
       "  {'armour': 3, 'cost': 53, 'damage': 0, 'name': 'Splintmail'},\n",
       "  {'armour': 4, 'cost': 75, 'damage': 0, 'name': 'Bandedmail'},\n",
       "  {'armour': 5, 'cost': 102, 'damage': 0, 'name': 'Platemail'},\n",
       "  {}],\n",
       " [{'armour': 0, 'cost': 25, 'damage': 1, 'name': 'Damage+1'},\n",
       "  {'armour': 0, 'cost': 50, 'damage': 2, 'name': 'Damage+2'},\n",
       "  {'armour': 0, 'cost': 100, 'damage': 3, 'name': 'Damage+3'},\n",
       "  {'armour': 1, 'cost': 20, 'damage': 0, 'name': 'Defense+1'},\n",
       "  {'armour': 2, 'cost': 40, 'damage': 0, 'name': 'Defense+2'},\n",
       "  {'armour': 3, 'cost': 80, 'damage': 0, 'name': 'Defense+3'},\n",
       "  {},\n",
       "  {}])"
      ]
     },
     "execution_count": 141,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "weapons = parse_table(weapons_s)\n",
    "armour = parse_table(armour_s)\n",
    "armour += [{}]\n",
    "rings = parse_table(rings_s)\n",
    "rings += [{}, {}]\n",
    "weapons, armour, rings"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 142,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def player(equipment):\n",
    "    player = {'hp': 100, 'damage': 0, 'armour': 0, 'cost': 0, 'name': ''}\n",
    "    for e in equipment:\n",
    "        for s in e:\n",
    "            player[s] += e[s]\n",
    "    return player"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 143,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'armour': 4, 'cost': 217, 'damage': 10, 'hp': 100, 'name': ''}"
      ]
     },
     "execution_count": 143,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "player([{'armour': 0, 'cost': 74, 'damage': 8}, {'armour': 1, 'cost': 13, 'damage': 0},\n",
    "        {'armour': 0, 'cost': 50, 'damage': 2}, {'armour': 3, 'cost': 80, 'damage': 0}])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 144,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "840"
      ]
     },
     "execution_count": 144,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "equipment = itertools.chain(\n",
    "    itertools.product(\n",
    "        itertools.combinations(weapons, 1),\n",
    "        itertools.combinations(armour, 1),\n",
    "        itertools.combinations(rings, 2)))\n",
    "\n",
    "players = [player(j for i in e for j in i) for e in equipment]\n",
    "len(players)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 146,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "91"
      ]
     },
     "execution_count": 146,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "winner = [p for p in players if not player_wins_fight(p, boss)]\n",
    "# sorted(winners, key=lambda w: w['cost'])\n",
    "min(winners, key=lambda w: w['cost'])['cost']"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#Part 2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 149,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "840"
      ]
     },
     "execution_count": 149,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "equipment = itertools.chain(\n",
    "    itertools.product(\n",
    "        itertools.combinations(weapons, 1),\n",
    "        itertools.combinations(armour, 1),\n",
    "        itertools.combinations(rings, 2)))\n",
    "\n",
    "players = [player(j for i in e for j in i) for e in equipment]\n",
    "len(players)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 151,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "158"
      ]
     },
     "execution_count": 151,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "losers = [p for p in players if not player_wins_fight(p, boss)]\n",
    "# sorted(losers, key=lambda w: w['cost'], reverse=True)\n",
    "max(losers, key=lambda w: w['cost'])['cost']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "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.4.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}