--- /dev/null
+{
+ "cells": [
+ {
+ "cell_type": "code",
+ "execution_count": 24,
+ "metadata": {
+ "collapsed": true
+ },
+ "outputs": [],
+ "source": [
+ "from collections import defaultdict\n",
+ "import itertools"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 1,
+ "metadata": {
+ "collapsed": false
+ },
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "['Frosting: capacity 4, durability -2, flavor 0, texture 0, calories 5',\n",
+ " 'Candy: capacity 0, durability 5, flavor -1, texture 0, calories 8',\n",
+ " 'Butterscotch: capacity -1, durability 0, flavor 5, texture 0, calories 6',\n",
+ " 'Sugar: capacity 0, durability 0, flavor -2, texture 2, calories 1']"
+ ]
+ },
+ "execution_count": 1,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "pi15 = [l.strip() for l in open('advent15.txt').readlines()]\n",
+ "pi15"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 9,
+ "metadata": {
+ "collapsed": false
+ },
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "{'Butterscotch': {'calories': 6,\n",
+ " 'capacity': -1,\n",
+ " 'durability': 0,\n",
+ " 'flavor': 5,\n",
+ " 'texture': 0},\n",
+ " 'Candy': {'calories': 8,\n",
+ " 'capacity': 0,\n",
+ " 'durability': 5,\n",
+ " 'flavor': -1,\n",
+ " 'texture': 0},\n",
+ " 'Frosting': {'calories': 5,\n",
+ " 'capacity': 4,\n",
+ " 'durability': -2,\n",
+ " 'flavor': 0,\n",
+ " 'texture': 0},\n",
+ " 'Sugar': {'calories': 1,\n",
+ " 'capacity': 0,\n",
+ " 'durability': 0,\n",
+ " 'flavor': -2,\n",
+ " 'texture': 2}}"
+ ]
+ },
+ "execution_count": 9,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "ingredients = {}\n",
+ "for l in pi15:\n",
+ " ls = l.split(': ')\n",
+ " name = ls[0]\n",
+ " props = ls[1].split(', ')\n",
+ " properties = {}\n",
+ " for p in props:\n",
+ " ps = p.split(' ')\n",
+ " properties[ps[0].strip()] = int(ps[1].strip())\n",
+ " ingredients[name] = properties\n",
+ "ingredients"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 33,
+ "metadata": {
+ "collapsed": false
+ },
+ "outputs": [],
+ "source": [
+ "def score(recipe, ingredients):\n",
+ " property_scores = defaultdict(int)\n",
+ " for ingredient, quantity in recipe:\n",
+ " for p in ingredients[ingredient]:\n",
+ " property_scores[p] += ingredients[ingredient][p] * quantity\n",
+ " total = 1\n",
+ " for p in property_scores:\n",
+ " total *= max(property_scores[p], 0)\n",
+ " return total"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 34,
+ "metadata": {
+ "collapsed": false
+ },
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "9396000000"
+ ]
+ },
+ "execution_count": 34,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "score([('Frosting', 30), ('Butterscotch', 30), ('Candy', 30), ('Sugar', 10)], ingredients)"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 31,
+ "metadata": {
+ "collapsed": false,
+ "scrolled": true
+ },
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "[(0, 0, 0, 3),\n",
+ " (0, 0, 1, 2),\n",
+ " (0, 0, 2, 1),\n",
+ " (0, 0, 3, 0),\n",
+ " (0, 1, 0, 2),\n",
+ " (0, 1, 1, 1),\n",
+ " (0, 1, 2, 0),\n",
+ " (0, 2, 0, 1),\n",
+ " (0, 2, 1, 0),\n",
+ " (0, 3, 0, 0),\n",
+ " (1, 0, 0, 2),\n",
+ " (1, 0, 1, 1),\n",
+ " (1, 0, 2, 0),\n",
+ " (1, 1, 0, 1),\n",
+ " (1, 1, 1, 0),\n",
+ " (1, 2, 0, 0),\n",
+ " (2, 0, 0, 1),\n",
+ " (2, 0, 1, 0),\n",
+ " (2, 1, 0, 0),\n",
+ " (3, 0, 0, 0)]"
+ ]
+ },
+ "execution_count": 31,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "capacity = 3\n",
+ "list(filter(lambda t: sum(t) == capacity,\n",
+ " itertools.product(range(capacity+1), range(capacity+1), \n",
+ " range(capacity+1), range(capacity+1))))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 38,
+ "metadata": {
+ "collapsed": false,
+ "scrolled": true
+ },
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "10618782720"
+ ]
+ },
+ "execution_count": 38,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "capacity = 100\n",
+ "max(score([('Frosting', f), ('Butterscotch', b), ('Candy', c), ('Sugar', s)], ingredients) \n",
+ " for b, c, f, s in filter(lambda t: sum(t) == capacity,\n",
+ " itertools.product(range(capacity+1), range(capacity+1), \n",
+ " range(capacity+1), range(capacity+1))))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 23,
+ "metadata": {
+ "collapsed": false
+ },
+ "outputs": [
+ {
+ "ename": "TypeError",
+ "evalue": "unhashable type: 'slice'",
+ "output_type": "error",
+ "traceback": [
+ "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
+ "\u001b[1;31mTypeError\u001b[0m Traceback (most recent call last)",
+ "\u001b[1;32m<ipython-input-23-95769c86abac>\u001b[0m in \u001b[0;36m<module>\u001b[1;34m()\u001b[0m\n\u001b[1;32m----> 1\u001b[1;33m \u001b[0mingredients\u001b[0m\u001b[1;33m[\u001b[0m\u001b[1;36m1\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m]\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m",
+ "\u001b[1;31mTypeError\u001b[0m: unhashable type: 'slice'"
+ ]
+ }
+ ],
+ "source": []
+ },
+ {
+ "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
+}
projects / advent-of-code-15.git / blobdiff