leetcode 227. 基本计算器 II

发表于 | 分类于

实现一个基本的计算器来计算一个简单的字符串表达式的值。

字符串表达式仅包含非负整数，+， - ，*，/ 四种运算符和空格 。 整数除法仅保留整数部分。

示例 1:

输入: "3+2*2"
输出: 7

示例 2:

输入: " 3/2 "
输出: 1

示例 3:

输入: " 3+5 / 2 "
输出: 5

说明：

  • 你可以假设所给定的表达式都是有效的。
  • 请不要使用内置的库函数 eval。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import re
import math
class Solution:
    def calculate(self, s):
        """
        :type s: str
        :rtype: int
        """
        s = s.replace(' ','')
        ops = [c for c in s if not c.isdigit()]
        numstr = re.split('[+-/*]',s)
        nums = [int(num) for num in numstr]
        stack = []
        stack.append(nums[0])
        i = 1
        j = 0
        while i < len(nums) and j < len(ops):
            stack.append(nums[i])
            if ops[j] == '-':
                num1 = stack.pop()
                stack.append(-num1)
            if ops[j] == '*':
                num1 = stack.pop()
                num2 = stack.pop()
                stack.append(num2*num1)
            if ops[j] == '/':
                num1 = stack.pop()
                num2 = stack.pop()
                t1 = num2/num1
                t2 = math.floor(t1) if t1 > 0 else math.ceil(t1)
                stack.append(t2)
            i += 1
            j += 1
        return sum(stack)

还有一个方法，使用正则来解决，但是消耗的时间更大