Python - Kalkulator 998, SPOJ, optymalizacja

Napisz program, który działa jak prosty kalkulator z pamięcią obsługujący pięć operacji: dodawanie, odejmowanie, mnożenie, dzielenie i obliczanie reszty z dzielenia liczb całkowitych. Przyjmujemy, że nasz kalkulator ma w pamięci 10 rejestrów o numerach 0 - 9 oraz że wszystkie rejestry zawierają na początku zera.

Wejście

Na wejście programu podana zostanie pewna nieokreślona liczba zestawów danych. Zestawy składają się z jednoznakowego symbolu operacji do wykonania (+, -, *, /, % i z) oraz następujących po nim dwóch liczb całkowitych.

Wyjście

Na wyjściu programu ma się pojawić ciąg liczb będących rezultatem wykonania pojawiających się na wejściu poleceń. Poszczególne liczby należy rozdzielić znakami nowej linii. To, co należy wykonać, zależy od symbolu operacji. Symbole +, -, *, / i % nakazują wykonanie odpowiednio dodawania, odejmowania, mnożenia, dzielenia i obliczenia reszty z dzielenia liczb, które są przechowywane w rejestrach o podanych numerach. Symbol z nakazuje zapis wartości podanej jako drugi argument w rejestrze o numerze podanym jako pierwszy argument. (Po wykonaniu tej operacji nie trzeba nic wyświetlać). Uwaga! Można założyć, że dane wejściowe nie zawierają polecenia dzielenia przez 0.

Przykład

Wejście:
z 3 6
z 1 89
z 2 60
z 0 11
+ 0 1
- 1 2
* 2 3
/ 3 0
% 3 1

Wyjście:
100
29
360
0
6 

#!/usr/bin/env python
# -*- coding: utf-8 -*-

rejestry = [0]*10
line = ""

try:
    line = input()
except EOFError:
    pass

# Empty sequences are false
while line:

    if line[0] == 'z':
        i = 4
        second = 0 # Length of second argument 
        while i < len(line):
            i += 1
            second += 1
        rejestry[int(line[2])] = int(line[4:4+second])

    elif line[0] == '+':
        a = int(rejestry[int(line[2])])
        b = int(rejestry[int(line[4])])
        print (str(a + b))
    elif line[0] == '-':
        a = int(rejestry[int(line[2])])
        b = int(rejestry[int(line[4])])
        print (str(a - b))
    elif line[0] == '*':
        a = int(rejestry[int(line[2])])
        b = int(rejestry[int(line[4])])
        print (str(a * b))
    elif line[0] == '/':
        a = int(rejestry[int(line[2])])
        b = int(rejestry[int(line[4])])
        print (str(int(a / b)))
    elif line[0] == '%':
        a = int(rejestry[int(line[2])])
        b = int(rejestry[int(line[4])])
        print (str(a % b))

    try:
        line = input()
    except EOFError:
        pass
 

        while i < len(line):
            i += 1
            second += 1

# -*- coding: utf-8 -*-
import sys

debug = False
if len(sys.argv) > 1:
	debug = True


def main():
	tab = [0 for x in range(10)]
	for linia in sys.stdin:
		znak, l1, l2 = linia.split()
		l1, l2 = int(l1), int(l2)
		if debug is True:
			print(znak, l1, l2)
		if znak == "z":
			tab[l1] = l2
		if znak == "+":
			print(int(tab[l1] + tab[l2]))
		if znak == "-":
			print(int(tab[l1] - tab[l2]))
		if znak == "*":
			print(int(tab[l1] * tab[l2]))
		if znak == "/":
			print(int(tab[l1] / tab[l2]))
		if znak == "%":
			print(int(tab[l1] % tab[l2]))
	return

main()

# -*- coding: utf-8 -*-
import sys

debug = False
if len(sys.argv) > 1:
	debug = True


def main():
	tab = ['0' for x in range(10)]
	for linia in sys.stdin:
		znak, l1, l2 = linia.split()
		l1, l2 = int(l1), int(l2)
		if debug is True:
			print(znak, l1, l2)
		if znak == "z":
			tab[l1] = str(l2)
		else:
			print(int(eval(tab[l1] + znak + tab[l2])))
	return

main()