147 lines
3.7 KiB
Plaintext
147 lines
3.7 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 7,
|
|
"metadata": {},
|
|
"outputs": [],
|
|
"source": [
|
|
"import numpy as np\n",
|
|
"import re\n",
|
|
"\n",
|
|
"def readfile(infilename='testinput',offset=0):\n",
|
|
" pattern= re.compile(r'''[XY][+=](\\d+)''') #Pattern to Extract Values\n",
|
|
" A=np.zeros((2,2),dtype=int) #Empty 2x2 Matrix\n",
|
|
" B=None #Placeholder for PrizePosition\n",
|
|
"\n",
|
|
" with open(infilename,'r') as infile:\n",
|
|
" for idx,line in enumerate(infile.readlines()):\n",
|
|
" line = line.strip()\n",
|
|
" # on empty line yield result and reset\n",
|
|
" if line =='':\n",
|
|
" if B is not None:\n",
|
|
" solution = np.linalg.solve(A.astype(int),B.astype(int))\n",
|
|
" # remove non integer Solutions\n",
|
|
" solution = solution.astype(int) if np.allclose(solution, np.round(solution)) else None\n",
|
|
" yield A,B,solution\n",
|
|
" A=np.zeros((2,2))\n",
|
|
" B=None\n",
|
|
" continue\n",
|
|
" # grep X/Y Values from Line\n",
|
|
" matches = pattern.findall(line)\n",
|
|
" if \"Button A\" in line: #A Button Presses are X in my equation > first column of A\n",
|
|
" A[:,0]=matches\n",
|
|
" if \"Button B\" in line: #B Button Presses are Y in my equation > second column of A\n",
|
|
" A[:,1]=matches\n",
|
|
" if \"Prize:\" in line:\n",
|
|
" B=np.array(matches,dtype=int)+[offset,offset] #offset added for part 2"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"Part 1"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 8,
|
|
"metadata": {},
|
|
"outputs": [
|
|
{
|
|
"data": {
|
|
"text/plain": [
|
|
"28714"
|
|
]
|
|
},
|
|
"execution_count": 8,
|
|
"metadata": {},
|
|
"output_type": "execute_result"
|
|
}
|
|
],
|
|
"source": [
|
|
"sum(a*3 + b for a,b in [x[2] for x in readfile('input') if x[2] is not None ])"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "markdown",
|
|
"metadata": {},
|
|
"source": [
|
|
"Part 2"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 225,
|
|
"metadata": {},
|
|
"outputs": [
|
|
{
|
|
"data": {
|
|
"text/plain": [
|
|
"[(array([[69., 27.],\n",
|
|
" [23., 71.]]),\n",
|
|
" array([10000000018641, 10000000010279]),\n",
|
|
" array([102851800151, 107526881786]))]"
|
|
]
|
|
},
|
|
"execution_count": 225,
|
|
"metadata": {},
|
|
"output_type": "execute_result"
|
|
}
|
|
],
|
|
"source": [
|
|
"[x for x in readfile(offset=10000000000000) if x[2] is not None ]\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": 226,
|
|
"metadata": {},
|
|
"outputs": [
|
|
{
|
|
"data": {
|
|
"text/plain": [
|
|
"416082282239"
|
|
]
|
|
},
|
|
"execution_count": 226,
|
|
"metadata": {},
|
|
"output_type": "execute_result"
|
|
}
|
|
],
|
|
"source": [
|
|
"sum(a*3 + b for a,b in [x[2] for x in readfile(offset=10000000000000) if x[2] is not None ])\n"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"metadata": {},
|
|
"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.8.10"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 2
|
|
}
|