"| `inc r` | increment contents of register `r` |\n",
"| `dec r` | decrement contents of register `r` |\n",
"| `set r i` | set contents of register `r` to literal value `i` |\n",
- "| `cpy r s` | set contents of register `r` into register `s` | \n",
+ "| `cpy r s` | copy contents of register `r` into register `s` | \n",
"| `sto r m` | store contents of register `r` into memory location `m` |\n",
"| `ld r m` | load contents of memory location `m` into register `r` |\n",
"| `jmp i` | jump to instruction `i` places forward |\n",
},
{
"cell_type": "code",
- "execution_count": 22,
+ "execution_count": 1,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 23,
+ "execution_count": 2,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 24,
+ "execution_count": 3,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 25,
+ "execution_count": 4,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 26,
+ "execution_count": 5,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 27,
+ "execution_count": 6,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 28,
+ "execution_count": 7,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 29,
+ "execution_count": 8,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 30,
+ "execution_count": 9,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 31,
+ "execution_count": 10,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 32,
+ "execution_count": 11,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 33,
+ "execution_count": 12,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 34,
+ "execution_count": 13,
"metadata": {},
"outputs": [
{
"{'a': 2, 'b': 1, 'c': 0, 'instructions': [], 'pc': 3}"
]
},
- "execution_count": 34,
+ "execution_count": 13,
"metadata": {},
"output_type": "execute_result"
}
},
{
"cell_type": "code",
- "execution_count": 35,
+ "execution_count": 14,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 36,
+ "execution_count": 15,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"def program_from_listing(listing, machine):\n",
- " labelled_instructions = [i for i in listing.split('\\n') if i]\n",
+ " labelled_instructions = [i.strip() for i in listing.split('\\n') \n",
+ " if i.strip() \n",
+ " if not i.strip().startswith('#')]\n",
" instructions = replace_labels(labelled_instructions)\n",
" program_from_instructions(instructions, machine)"
]
},
{
"cell_type": "code",
- "execution_count": 37,
+ "execution_count": 16,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 38,
+ "execution_count": 17,
"metadata": {},
"outputs": [
{
"['inc', 'a']"
]
},
- "execution_count": 38,
+ "execution_count": 17,
"metadata": {},
"output_type": "execute_result"
}
},
{
"cell_type": "code",
- "execution_count": 39,
+ "execution_count": 18,
"metadata": {},
"outputs": [
{
- "data": {
- "text/plain": [
- "['set a 10', 'dec a', 'inc b', 'sto b 1', 'jpz a 2', 'jmp -4']"
- ]
- },
- "execution_count": 39,
- "metadata": {},
- "output_type": "execute_result"
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "set a 10\n",
+ "dec a\n",
+ "inc b\n",
+ "sto b 1\n",
+ "jpz a 2\n",
+ "jmp -4\n"
+ ]
}
],
"source": [
"program = \"\"\"\n",
" set a 10\n",
+ " # comment line\n",
+ " \n",
"loop: dec a\n",
" inc b\n",
" sto b 1\n",
" jpz a 2\n",
" jmp loop\n",
"\"\"\"\n",
- "labelled_instructions = [i for i in program.split('\\n') if i]\n",
+ "labelled_instructions = [i.strip() for i in program.split('\\n') if i.strip() if not i.strip().startswith('#')]\n",
"instructions = replace_labels(labelled_instructions)\n",
- "instructions"
+ "print('\\n'.join(instructions))"
]
},
{
"cell_type": "code",
- "execution_count": 40,
+ "execution_count": 19,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 41,
+ "execution_count": 20,
"metadata": {
"collapsed": true
},
},
{
"cell_type": "code",
- "execution_count": 42,
+ "execution_count": 21,
"metadata": {},
"outputs": [
{
" 'pc': 4}"
]
},
- "execution_count": 42,
+ "execution_count": 21,
"metadata": {},
"output_type": "execute_result"
}
},
{
"cell_type": "code",
- "execution_count": 43,
+ "execution_count": 22,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
- "{'b': 10,\n",
- " 1: 10,\n",
+ "{1: 10,\n",
+ " 'a': 0,\n",
+ " 'b': 10,\n",
" 'instructions': [(<function __main__.set_literal>, ['a', 10]),\n",
" (<function __main__.dec>, ['a']),\n",
" (<function __main__.inc>, ['b']),\n",
" (<function __main__.sto>, ['b', 1]),\n",
" (<function __main__.jpz>, ['a', 2]),\n",
" (<function __main__.jmp>, [-4])],\n",
- " 'pc': 6,\n",
- " 'a': 0,\n",
- " 'c': 20}"
+ " 'c': 20,\n",
+ " 'pc': 6}"
]
},
- "execution_count": 43,
+ "execution_count": 22,
"metadata": {},
"output_type": "execute_result"
}
},
{
"cell_type": "code",
- "execution_count": 44,
+ "execution_count": 23,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
- "{'b': 10,\n",
- " 1: 10,\n",
+ "{1: 10,\n",
+ " 'a': 0,\n",
+ " 'b': 10,\n",
" 'instructions': [(<function __main__.set_literal>, ['a', 10]),\n",
" (<function __main__.dec>, ['a']),\n",
" (<function __main__.inc>, ['b']),\n",
" (<function __main__.sto>, ['b', 1]),\n",
" (<function __main__.jpz>, ['a', 2]),\n",
" (<function __main__.jmp>, [-4])],\n",
- " 'pc': 6,\n",
- " 'a': 0,\n",
- " 'c': 20}"
+ " 'c': 20,\n",
+ " 'pc': 6}"
]
},
- "execution_count": 44,
+ "execution_count": 23,
"metadata": {},
"output_type": "execute_result"
}
},
{
"cell_type": "code",
- "execution_count": 45,
+ "execution_count": 24,
"metadata": {},
"outputs": [
{
" 'pc': 10}"
]
},
- "execution_count": 45,
+ "execution_count": 24,
"metadata": {},
"output_type": "execute_result"
}
},
{
"cell_type": "code",
- "execution_count": 49,
+ "execution_count": 25,
"metadata": {},
"outputs": [
{
" 'pc': 9}"
]
},
- "execution_count": 49,
+ "execution_count": 25,
"metadata": {},
"output_type": "execute_result"
}
},
{
"cell_type": "code",
- "execution_count": 52,
+ "execution_count": 26,
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
- "'10: 7, a: 0, b: 1, c: 3, pc: 11'"
+ "'1: 8, a: 0, b: 1, c: 8, pc: 11'"
]
},
- "execution_count": 52,
+ "execution_count": 26,
"metadata": {},
"output_type": "execute_result"
}
"source": [
"# c holds floor(a/2)\n",
"program = \"\"\"\n",
- " sto a 10\n",
" set c 0\n",
" set b 0\n",
"loop: dec a\n",
"odd: inc b\n",
"end: jpz a 2\n",
" jmp loop\n",
+ " sto c 1\n",
+ "\"\"\"\n",
+ "# m = new_machine()\n",
+ "# program_from_listing(program, m)\n",
+ "# run(m)\n",
+ "show_machine(execute(program, initial_state={'a': 17}))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 27,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "'a: 4, b: 0, c: 12, pc: 9'"
+ ]
+ },
+ "execution_count": 27,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "# c holds a * 3\n",
+ "program = \"\"\"\n",
+ " set c 0\n",
+ " cpy a b\n",
+ " # start of main loop\n",
+ "loop: jpz b end\n",
+ " dec b\n",
+ " inc c\n",
+ " inc c\n",
+ " inc c\n",
+ " jmp loop\n",
+ " \n",
+ " # end of program \n",
+ " \n",
+ "end: jmp 1\n",
+ "\"\"\"\n",
+ "# m = new_machine()\n",
+ "# program_from_listing(program, m)\n",
+ "# run(m)\n",
+ "show_machine(execute(program, initial_state={'a': 4}))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 28,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "'1: 0, a: 0, b: 0, c: 27, pc: 11'"
+ ]
+ },
+ "execution_count": 28,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "# c holds a * b\n",
+ "program = \"\"\"\n",
+ " set c 0\n",
+ " sto a 1\n",
+ "loop: jpz b end \n",
+ " dec b\n",
+ " ld a 1\n",
+ "smul: jpz a emul\n",
+ " inc c\n",
+ " dec a\n",
+ " jmp smul\n",
+ "emul: jmp loop \n",
+ " \n",
+ "end: sto a 1\n",
+ "\"\"\"\n",
+ "m = new_machine()\n",
+ "program_from_listing(program, m)\n",
+ "run(m)\n",
+ "show_machine(execute(program, initial_state={'a': 9, 'b': 3}))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 29,
+ "metadata": {},
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "set c 0\n",
+ "sto a 1\n",
+ "jpz b 8\n",
+ "dec b\n",
+ "ld a 1\n",
+ "jpz a 4\n",
+ "inc c\n",
+ "dec a\n",
+ "jmp -3\n",
+ "jmp -7\n",
+ "sto a 1\n"
+ ]
+ }
+ ],
+ "source": [
+ "labelled_instructions = [i.strip() for i in program.split('\\n') if i.strip() if not i.strip().startswith('#')]\n",
+ "instructions = replace_labels(labelled_instructions)\n",
+ "print('\\n'.join(instructions))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 30,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "'1: 0, a: 0, b: 0, c: 10, pc: 13'"
+ ]
+ },
+ "execution_count": 30,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "# c holds a * b\n",
+ "program = \"\"\"\n",
+ " set a 2\n",
+ " set b 5\n",
+ " set c 0\n",
+ " sto a 1\n",
+ "loop: jpz b end \n",
+ " dec b\n",
+ " ld a 1\n",
+ "smul: jpz a emul\n",
+ " inc c\n",
+ " dec a\n",
+ " jmp smul\n",
+ "emul: jmp loop \n",
+ " \n",
+ "end: sto a 1\n",
+ "\"\"\"\n",
+ "m = new_machine()\n",
+ "program_from_listing(program, m)\n",
+ "run(m)\n",
+ "show_machine(execute(program, initial_state={'a': 9, 'b': 3}))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 31,
+ "metadata": {},
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "set a 2\n",
+ "set b 5\n",
+ "set c 0\n",
+ "sto a 1\n",
+ "jpz b 8\n",
+ "dec b\n",
+ "ld a 1\n",
+ "jpz a 4\n",
+ "inc c\n",
+ "dec a\n",
+ "jmp -3\n",
+ "jmp -7\n",
+ "sto a 1\n"
+ ]
+ }
+ ],
+ "source": [
+ "labelled_instructions = [i.strip() for i in program.split('\\n') \n",
+ " if i.strip() \n",
+ " if not i.strip().startswith('#')]\n",
+ "instructions = replace_labels(labelled_instructions)\n",
+ "print('\\n'.join(instructions))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 32,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "'1: 0, a: 52, b: 0, c: 0, pc: 48'"
+ ]
+ },
+ "execution_count": 32,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "# Collatz. a initially holds value, but location 1 is used to store the current value as we're going along.\n",
+ "# Location 2 holds number of steps taken\n",
+ "# Location 3 holds max value reached\n",
+ "program = \"\"\"\n",
+ " sto a 1\n",
+ " \n",
+ " set b 0\n",
+ " \n",
+ " # if a is one, finish, \n",
+ "main: dec a\n",
+ " jpz a end\n",
+ " inc a\n",
+ " \n",
+ " # find parity of a\n",
+ " cpy a c\n",
+ " set b 0\n",
+ "prty: dec c\n",
+ " jpz b odd\n",
+ " dec b\n",
+ " jmp prte\n",
+ "odd: inc b\n",
+ "prte: jpz c 2\n",
+ " jmp prty\n",
+ " \n",
+ " # b == 0 means a even; b == 1 means a odd\n",
+ " jpz b isev\n",
+ "\n",
+ " # c holds a * 3 + 1\n",
+ " cpy a b\n",
+ "mul: jpz b emul\n",
+ " dec b\n",
+ " inc c\n",
+ " inc c\n",
+ " inc c\n",
+ " jmp mul\n",
+ "emul: inc c\n",
+ " cpy c a\n",
+ " jmp fin\n",
+ " \n",
+ " \n",
+ "isev: set c 0\n",
+ " set b 0\n",
+ "hlvl: dec a\n",
+ " jpz b oddh\n",
+ " dec b\n",
+ " inc c\n",
+ " jmp endh\n",
+ "oddh: inc b\n",
+ "endh: jpz a 2\n",
+ " jmp hlvl\n",
+ " cpy c a\n",
+ "\n",
+ "fin: cpy a b\n",
+ " ld c 1\n",
+ "maxc: jpz c this\n",
+ " jpz b othr\n",
+ " dec b\n",
+ " dec c\n",
+ " jmp maxc\n",
+ "this: sto a 1\n",
+ "othr: jmp main\n",
+ " \n",
+ " # end of program \n",
+ " \n",
+ "end: ld a 1\n",
+ " set c 0\n",
+ " sto c 1\n",
"\"\"\"\n",
"# m = new_machine()\n",
"# program_from_listing(program, m)\n",
"show_machine(execute(program, initial_state={'a': 7}))"
]
},
+ {
+ "cell_type": "code",
+ "execution_count": 33,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "40"
+ ]
+ },
+ "execution_count": 33,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "13*3+1"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 34,
+ "metadata": {
+ "collapsed": true
+ },
+ "outputs": [],
+ "source": [
+ "def max_collatz(start):\n",
+ " mc = start\n",
+ " i = start\n",
+ " while i != 1:\n",
+ " if i % 2 == 0:\n",
+ " i = i // 2\n",
+ " else:\n",
+ " i = 3 * i + 1\n",
+ " if i > mc:\n",
+ " mc = i\n",
+ " return mc"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 35,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "52"
+ ]
+ },
+ "execution_count": 35,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "max_collatz(7)"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 36,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "(250504, 937)"
+ ]
+ },
+ "execution_count": 36,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "max([(max_collatz(i), i) for i in range(1, 1000)])"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 41,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "(52, 9)"
+ ]
+ },
+ "execution_count": 41,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "max([(max_collatz(i), i) for i in range(1, 10)])"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 37,
+ "metadata": {
+ "scrolled": true
+ },
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "[(1, 0, 1),\n",
+ " (2, 0, 2),\n",
+ " (3, 0, 16),\n",
+ " (4, 0, 4),\n",
+ " (5, 0, 16),\n",
+ " (6, 0, 16),\n",
+ " (7, 0, 52),\n",
+ " (8, 0, 8),\n",
+ " (9, 0, 52),\n",
+ " (10, 0, 16),\n",
+ " (11, 0, 52),\n",
+ " (12, 0, 16),\n",
+ " (13, 0, 40),\n",
+ " (14, 0, 52),\n",
+ " (15, 0, 160),\n",
+ " (16, 0, 16),\n",
+ " (17, 0, 52),\n",
+ " (18, 0, 52),\n",
+ " (19, 0, 88),\n",
+ " (20, 0, 20),\n",
+ " (21, 0, 64),\n",
+ " (22, 0, 52),\n",
+ " (23, 0, 160),\n",
+ " (24, 0, 24),\n",
+ " (25, 0, 88),\n",
+ " (26, 0, 40),\n",
+ " (27, 0, 9232),\n",
+ " (28, 0, 52),\n",
+ " (29, 0, 88),\n",
+ " (30, 0, 160),\n",
+ " (31, 0, 9232),\n",
+ " (32, 0, 32),\n",
+ " (33, 0, 100),\n",
+ " (34, 0, 52),\n",
+ " (35, 0, 160),\n",
+ " (36, 0, 52),\n",
+ " (37, 0, 112),\n",
+ " (38, 0, 88),\n",
+ " (39, 0, 304),\n",
+ " (40, 0, 40),\n",
+ " (41, 0, 9232),\n",
+ " (42, 0, 64),\n",
+ " (43, 0, 196),\n",
+ " (44, 0, 52),\n",
+ " (45, 0, 136),\n",
+ " (46, 0, 160),\n",
+ " (47, 0, 9232),\n",
+ " (48, 0, 48),\n",
+ " (49, 0, 148),\n",
+ " (50, 0, 88),\n",
+ " (51, 0, 232),\n",
+ " (52, 0, 52),\n",
+ " (53, 0, 160),\n",
+ " (54, 0, 9232),\n",
+ " (55, 0, 9232),\n",
+ " (56, 0, 56),\n",
+ " (57, 0, 196),\n",
+ " (58, 0, 88),\n",
+ " (59, 0, 304),\n",
+ " (60, 0, 160),\n",
+ " (61, 0, 184),\n",
+ " (62, 0, 9232),\n",
+ " (63, 0, 9232),\n",
+ " (64, 0, 64),\n",
+ " (65, 0, 196),\n",
+ " (66, 0, 100),\n",
+ " (67, 0, 304),\n",
+ " (68, 0, 68),\n",
+ " (69, 0, 208),\n",
+ " (70, 0, 160),\n",
+ " (71, 0, 9232),\n",
+ " (72, 0, 72),\n",
+ " (73, 0, 9232),\n",
+ " (74, 0, 112),\n",
+ " (75, 0, 340),\n",
+ " (76, 0, 88),\n",
+ " (77, 0, 232),\n",
+ " (78, 0, 304),\n",
+ " (79, 0, 808),\n",
+ " (80, 0, 80),\n",
+ " (81, 0, 244),\n",
+ " (82, 0, 9232),\n",
+ " (83, 0, 9232),\n",
+ " (84, 0, 84),\n",
+ " (85, 0, 256),\n",
+ " (86, 0, 196),\n",
+ " (87, 0, 592),\n",
+ " (88, 0, 88),\n",
+ " (89, 0, 304),\n",
+ " (90, 0, 136),\n",
+ " (91, 0, 9232),\n",
+ " (92, 0, 160),\n",
+ " (93, 0, 280),\n",
+ " (94, 0, 9232),\n",
+ " (95, 0, 9232),\n",
+ " (96, 0, 96),\n",
+ " (97, 0, 9232),\n",
+ " (98, 0, 148),\n",
+ " (99, 0, 448)]"
+ ]
+ },
+ "execution_count": 37,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "ans = []\n",
+ "for i in range(1, 100):\n",
+ " m = execute(program, initial_state={'a': i})\n",
+ " c = max_collatz(i)\n",
+ " if m[1] != c:\n",
+ " ans += [(i, m[1], c)]\n",
+ "ans"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 38,
+ "metadata": {},
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "'1: 0, a: 250504, b: 0, c: 0, pc: 48'"
+ ]
+ },
+ "execution_count": 38,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
+ "source": [
+ "show_machine(execute(program, initial_state={'a': 937}))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 40,
+ "metadata": {},
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "1 loop, best of 3: 38.6 s per loop\n"
+ ]
+ }
+ ],
+ "source": [
+ "%%timeit\n",
+ "show_machine(execute(program, initial_state={'a': 937}))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 39,
+ "metadata": {
+ "collapsed": true
+ },
+ "outputs": [],
+ "source": [
+ "# labelled_instructions = [i.strip() for i in program.split('\\n') \n",
+ "# if i.strip() \n",
+ "# if not i.strip().startswith('#')]\n",
+ "# instructions = replace_labels(labelled_instructions)\n",
+ "# print('\\n'.join(instructions))\n",
+ "# open('07-program.txt', 'w').write('\\n'.join(instructions))"
+ ]
+ },
{
"cell_type": "code",
"execution_count": null,
projects / ou-summer-of-code-2017.git / blobdiff