leetcode如何實現基本計算器

這篇文章給大家分享的是有關leetcode如何實現基本計算器的內容。小編覺得挺實用的，因此分享給大家做個參考，一起跟隨小編過來看看吧。

一、題目內容

實現一個基本的計算器來計算一個簡單的字符串表達式 s 的值。

示例 1：

輸入：s = "1 + 1"
輸出：2

示例 2：

輸入：s = " 2-1 + 2 "
輸出：3

示例 3：

輸入：s = "(1+(4+5+2)-3)+(6+8)"
輸出：23
 

提示：

1 <= s.length <= 3 * 10^5
s 由數字、'+'、'-'、'('、')'、和 ' ' 組成
s 表示一個有效的表達式

二、解題思路

用棧來做，當遇到左括號時，存入之前的計算結果，再存入一個flag為1；

如果遇到右括號，則出棧flag為1并乘以當前括號內的計算結果，然后再加上出棧的之前的計算結果；

遇到數字，就計算與之后數字的構成的二位數，否則乘以 flag并與之前的計算結果相加更新計算結果；

三、代碼

class Solution:
    def calculate(self, s: str) -> int:
        stack = []
        res = 0
        flag = 1
        i = 0
        while i < len(s):
            if s[i] == '+':
                flag = 1
            elif s[i] == '-':
                flag = -1
            elif s[i] == '(':
                stack.append(res)
                res = 0
                stack.append(flag)
                flag = 1
            elif s[i] == ')':
                res = stack.pop() * res + stack.pop()
            elif s[i].isdigit():
                s_num = ord(s[i]) - ord('0')
                while i + 1 < len(s):
                    if s[i + 1].isdigit():
                        s_num = s_num * 10 + ord(s[i + 1]) - ord('0')
                        i += 1
                    else:
                        break

                res = res + flag * s_num
            i += 1

        return res


if __name__ == '__main__':
    ss = Solution()
    s = "(1+(4+5+2)-3)+(6+8)"
    # s = "2147483647"
    # s = "1 + 1"
    ans = ss.calculate(s)
    print(ans)

感謝各位的閱讀！關于“leetcode如何實現基本計算器”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，讓大家可以學到更多知識，如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！