逆波兰式Python版本

逆波兰式Python版本

算法描述

将一个普通的中序表达式转换为逆波兰表达式的一般算法是：
首先需要分配2个栈，一个作为临时存储运算符的栈S1（含一个结束符号），一个作为输入逆波兰式的栈S2（空栈），S1栈可先放入优先级最低的运算符#，注意，中缀式应以此最低优先级的运算符结束。可指定其他字符，不一定非#不可。从中缀式的左端开始取字符，逐序进行如下步骤：
（1）若取出的字符是操作数，则分析出完整的运算数，该操作数直接送入S2栈
（2）若取出的字符是运算符，则将该运算符与S1栈栈顶元素比较，如果该运算符优先级大于S1栈栈顶运算符优先级，则将该运算符进S1栈，否则，将S1栈的栈顶运算符弹出，送入S2栈中，直至S1栈栈顶运算符低于（不包括等于）该运算符优先级，最后将该运算符送入S1栈。
（3）若取出的字符是“（”，则直接送入S1栈顶。
（4）若取出的字符是“）”，则将距离S1栈栈顶最近的“（”之间的运算符，逐个出栈，依次送入S2栈，此时抛弃“（”。
（5）重复上面的1~4步，直至处理完所有的输入字符
（6）若取出的字符是“#”，则将S1栈内所有运算符（不包括“#”），逐个出栈，依次送入S2栈。
完成以上步骤，S2栈便为逆波兰式输出结果。不过S2应做一下逆序处理。便可以按照逆波兰式的计算方法计算了！

Python代码

# -*- coding: utf-8 -*-
#coding=utf-8
data = [ i for i in "1234567890"]
cal = {"+":1,"-":1,"*":2,"/":2}
cal1 = {"(":0}
def parse(source):
    result = []
    c = []
    slist = [i for i in source]
    for item in slist:
        if item in data:
            result.append(item)
        elif not c and item in cal.keys():
            c.append(item)
            continue
        elif c and item in cal.keys():
            for x in range(c.__len__()):
                z=c[-1]
                temp = cal[z] if cal.has_key(z) else cal1[z]
                if temp >= cal[item]:
                    result.append(c.pop())
                else:
                    c.append(item)
                    break
            if not c:
                c.append(item)
        elif item == ")":
            for x in range(c.__len__()):
                if c[-1] == "(":
                    c.pop()
                    break
                else:
                    result.append(c.pop())
        elif item == "(":
            c.append(item)
        #print(result,c)
    for x in range(c.__len__()):
        result.append(c.pop())
    return result

def caculate(source):
    num = []
    for i in source:
        if i in data:
            num.append(i)
        else:
            num1=num.pop()
            num2=num.pop()
            num.append(str(eval("%s%s%s"%(num2,i,num1))))
    return num[0]

if __name__ == "__main__":
    sourcestr = "5+(3-1)*2-6/2"
    print("orig:%s"%sourcestr)
    re1 = parse(sourcestr)
    print("niBolan:%s"%" ".join(re1))
    result = caculate(re1)
    print('result:%s'%result)