【华为机试真题 Python实现】仿 LISP 运算【2022 Q1 Q2 |200分】

题目描述

LISP 语言唯一的语法就是括号要配对。

形如 (OP P1 P2 …)，括号内元素由单个空格分割。

其中第一个元素 OP 为操作符，后续元素均为其参数，参数个数取决于操作符类型。

注意：

参数 P1, P2 也有可能是另外一个嵌套的 (OP P1 P2 …) ，

当前 OP 类型为 add / sub / mul / div（全小写），分别代表整数的加减乘除法，

简单起见，所有 OP 参数个数均为 2 。

举例:

输入：(mul 3 -7)

输出：21

输入：(add 1 2)

输出：3

输入：(sub (mul 2 4) (div 9 3))

输出：5

输入：(div 1 0)

输出：error

题目涉及数字均为整数，可能为负；

不考虑 32 位溢出翻转，计算过程中也不会发生 32 位溢出翻转，

除零错误时，输出 “error”，

除法遇除不尽，向下取整，即 3/2 = 1

def lisp(string_words):
    string_words = string_words.split()
    if string_words[0] == "add":
        del string_words[0]
        return sum([int(i) for i in string_words])
    if string_words[0] == "sub":
        return int(string_words[1]) - int(string_words[2])
    if string_words[0] == "mul":
        result = 1
        for i in range(1, len(string_words)):
            result *= int(string_words[i])
        return result
    if string_words[0] == "div":
        if string_words[2] == "0":
            return "error"
        else:
            return round(int(string_words[1]) // int(string_words[2]))
n = input()
while "(" in n:
    k = ''.join(n[::-1].split("(")[0])[::-1].split((")"))[0]
    n = n.replace(f'({k})', str(lisp(k)))
print(n)