PAT 02-3. 求前缀表达式的值（25） Python

02-线性结构3. 求前缀表达式的值（25） Python

题目要求

算术表达式有前缀表示法、中缀表示法和后缀表示法等形式。前缀表达式指二元运算符位于两个运算数之前，例如2+3*(7-4)+8/4的前缀表达式是：+ + 2 * 3 - 7 4 / 8 4。请设计程序计算前缀表达式的结果值。

输入格式说明：

输入在一行内给出不超过30个字符的前缀表达式，只包含+、-、*、\以及运算数，不同对象（运算数、运算符号）之间以空格分隔。

输出格式说明：

输出前缀表达式的运算结果，精确到小数点后1位，或错误信息“ERROR”。

样例输入与输出：

序号	输入	输出
1	+ + 2 * 3 - 7 4 / 8 4	13.0
2	/ -25 + * - 2 3 4 / 8 4	12.5
3	/ 5 + * - 2 3 4 / 8 2	ERROR
4	+ 10.23	10.2

分析：

解决这个问题的关键是：设置堆栈，从后向前扫描，遇到数字就入栈，遇到操作符就pop栈顶两个元素进行运算操作最后把结果再压栈。

实现说明：

本程序使用Python语言编写，但是存在不足之处，在测试中有一个测试点显示“返回非零”，既不完全正确，请斟酌学习。

实现代码：

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

'''
By:jixiangrurui
Date:2015.08.17
Nothing replaces hard work.
'''
def Operator(num1,num2,operator):
    '''(str, str, str) -> str

    Return the value of the expression.

    >>> Operator('1.5','2','/')
    '0.75'
    '''
    if operator == '+':
        result = float(num1) + float(num2)
    elif operator == '-':
        result = float(num1) - float(num2)
    elif operator == '*':
        result = float(num1) * float(num2)
    else:
        if num2 == '0.0':
            return 'ERROR'       
        result = float(num1) / float(num2)

    return str(result)


def DealExpression(strs):
    '''(list of str) -> str

    '''
    stack = []
    operator = ['+','-','*','/']
    length = len(strs)
    if length == 1:
        result = strs[0]
    elif length == 2:
        result = Operator('0',strs[1],strs[0])
    else:
        for i in range(length-1,-1,-1):
            if strs[i] not in operator:
                stack.append(strs[i])
            if strs[i] in operator:
                num1 = stack.pop()
                num2 = stack.pop()
                result = Operator(num1,num2,strs[i])
                if result == 'ERROR':
                    return result
                stack.append(result)
        result = stack[0]

    return result


if __name__ == '__main__':

    strs = input()
    strs = strs.split(" ")
    result = DealExpression(strs)
    if result == 'ERROR':
        print('ERROR')
    print("%.1f"%float(result))