讀完本文,你可以去力扣拿下如下題目:
-----------
我們最終要實現的計算器功能如下:
1、輸入一個字符串,可以包含+ - * /
、數字、括號以及空格,你的算法返回運算結果。
2、要符合運算法則,括號的優先級最高,先乘除后加減。
3、除號是整數除法,無論正負都向 0 取整(5/2=2,-5/2=-2)。
4、可以假定輸入的算式一定合法,且計算過程不會出現整型溢出,不會出現除數為 0 的意外情況。
比如輸入如下字符串,算法會返回 9:
3 * (2-6 /(3 -7))
可以看到,這就已經非常接近我們實際生活中使用的計算器了,雖然我們以前肯定都用過計算器,但是如果簡單思考一下其算法實現,就會大驚失色:
1、按照常理處理括號,要先計算最內層的括號,然后向外慢慢化簡。這個過程我們手算都容易出錯,何況寫成算法呢!
2、要做到先乘除,后加減,這一點教會小朋友還不算難,但教給計算機恐怕有點困難。
3、要處理空格。我們為了美觀,習慣性在數字和運算符之間打個空格,但是計算之中得想辦法忽略這些空格。
我記得很多大學數據結構的教材上,在講棧這種數據結構的時候,應該都會用計算器舉例,但是有一說一,講的真的垃圾,不知道多少未來的計算機科學家就被這種簡單的數據結構勸退了。
那么本文就來聊聊怎么實現上述一個功能完備的計算器功能,關鍵在於層層拆解問題,化整為零,逐個擊破,相信這種思維方式能幫大家解決各種復雜問題。
下面就來拆解,從最簡單的一個問題開始。
一、字符串轉整數
是的,就是這么一個簡單的問題,首先告訴我,怎么把一個字符串形式的正整數,轉化成 int 型?
string s = "458";
int n = 0;
for (int i = 0; i < s.size(); i++) {
char c = s[i];
n = 10 * n + (c - '0');
}
// n 現在就等於 458
這個還是很簡單的吧,老套路了。但是即便這么簡單,依然有坑:(c - '0')
的這個括號不能省略,否則可能造成整型溢出。
因為變量c
是一個 ASCII 碼,如果不加括號就會先加后減,想象一下s
如果接近 INT_MAX,就會溢出。所以用括號保證先減后加才行。
二、處理加減法
現在進一步,如果輸入的這個算式只包含加減法,而且不存在空格,你怎么計算結果?我們拿字符串算式1-12+3
為例,來說一個很簡單的思路:
1、先給第一個數字加一個默認符號+
,變成+1-12+3
。
2、把一個運算符和數字組合成一對兒,也就是三對兒+1
,-12
,+3
,把它們轉化成數字,然后放到一個棧中。
3、將棧中所有的數字求和,就是原算式的結果。
PS:我認真寫了 100 多篇原創,手把手刷 200 道力扣題目,全部發布在 labuladong的算法小抄,持續更新。建議收藏,按照我的文章順序刷題,掌握各種算法套路后投再入題海就如魚得水了。
我們直接看代碼,結合一張圖就看明白了:
int calculate(string s) {
stack<int> stk;
// 記錄算式中的數字
int num = 0;
// 記錄 num 前的符號,初始化為 +
char sign = '+';
for (int i = 0; i < s.size(); i++) {
char c = s[i];
// 如果是數字,連續讀取到 num
if (isdigit(c))
num = 10 * num + (c - '0');
// 如果不是數字,就是遇到了下一個符號,
// 之前的數字和符號就要存進棧中
if (!isdigit(c) || i == s.size() - 1) {
switch (sign) {
case '+':
stk.push(num); break;
case '-':
stk.push(-num); break;
}
// 更新符號為當前符號,數字清零
sign = c;
num = 0;
}
}
// 將棧中所有結果求和就是答案
int res = 0;
while (!stk.empty()) {
res += stk.top();
stk.pop();
}
return res;
}
我估計就是中間帶switch
語句的部分有點不好理解吧,i
就是從左到右掃描,sign
和num
跟在它身后。當s[i]
遇到一個運算符時,情況是這樣的:
所以說,此時要根據sign
的 case 不同選擇nums
的正負號,存入棧中,然后更新sign
並清零nums
記錄下一對兒符合和數字的組合。
另外注意,不只是遇到新的符號會觸發入棧,當i
走到了算式的盡頭(i == s.size() - 1
),也應該將前面的數字入棧,方便后續計算最終結果。
至此,僅處理緊湊加減法字符串的算法就完成了,請確保理解以上內容,后續的內容就基於這個框架修修改改就完事兒了。
三、處理乘除法
其實思路跟僅處理加減法沒啥區別,拿字符串2-3*4+5
舉例,核心思路依然是把字符串分解成符號和數字的組合。
比如上述例子就可以分解為+2
,-3
,*4
,+5
幾對兒,我們剛才不是沒有處理乘除號嗎,很簡單,其他部分都不用變,在switch
部分加上對應的 case 就行了:
for (int i = 0; i < s.size(); i++) {
char c = s[i];
if (isdigit(c))
num = 10 * num + (c - '0');
if (!isdigit(c) || i == s.size() - 1) {
switch (sign) {
int pre;
case '+':
stk.push(num); break;
case '-':
stk.push(-num); break;
// 只要拿出前一個數字做對應運算即可
case '*':
pre = stk.top();
stk.pop();
stk.push(pre * num);
break;
case '/':
pre = stk.top();
stk.pop();
stk.push(pre / num);
break;
}
// 更新符號為當前符號,數字清零
sign = c;
num = 0;
}
}
乘除法優先於加減法體現在,乘除法可以和棧頂的數結合,而加減法只能把自己放入棧。
現在我們思考一下如何處理字符串中可能出現的空格字符。其實也非常簡單,想想空格字符的出現,會影響我們現有代碼的哪一部分?
// 如果 c 非數字
if (!isdigit(c) || i == s.size() - 1) {
switch (c) {...}
sign = c;
num = 0;
}
顯然空格會進入這個 if 語句,但是我們並不想讓空格的情況進入這個 if,因為這里會更新sign
並清零nums
,空格根本就不是運算符,應該被忽略。
那么只要多加一個條件即可:
if ((!isdigit(c) && c != ' ') || i == s.size() - 1) {
...
}
好了,現在我們的算法已經可以按照正確的法則計算加減乘除,並且自動忽略空格符,剩下的就是如何讓算法正確識別括號了。
四、處理括號
處理算式中的括號看起來應該是最難的,但真沒有看起來那么難。
為了規避編程語言的繁瑣細節,我把前面解法的代碼翻譯成 Python 版本:
def calculate(s: str) -> int:
def helper(s: List) -> int:
stack = []
sign = '+'
num = 0
while len(s) > 0:
c = s.pop(0)
if c.isdigit():
num = 10 * num + int(c)
if (not c.isdigit() and c != ' ') or len(s) == 0:
if sign == '+':
stack.append(num)
elif sign == '-':
stack.append(-num)
elif sign == '*':
stack[-1] = stack[-1] * num
elif sign == '/':
# python 除法向 0 取整的寫法
stack[-1] = int(stack[-1] / float(num))
num = 0
sign = c
return sum(stack)
# 需要把字符串轉成列表方便操作
return helper(list(s))
這段代碼跟剛才 C++ 代碼完全相同,唯一的區別是,不是從左到右遍歷字符串,而是不斷從左邊pop
出字符,本質還是一樣的。
那么,為什么說處理括號沒有看起來那么難呢,因為括號具有遞歸性質。我們拿字符串3*(4-5/2)-6
舉例:
calculate(3*(4-5/2)-6
)
= 3 * calculate(4-5/2
) - 6
= 3 * 2 - 6
= 0
可以腦補一下,無論多少層括號嵌套,通過 calculate 函數遞歸調用自己,都可以將括號中的算式化簡成一個數字。換句話說,括號包含的算式,我們直接視為一個數字就行了。
現在的問題是,遞歸的開始條件和結束條件是什么?遇到(
開始遞歸,遇到)
結束遞歸:
def calculate(s: str) -> int:
def helper(s: List) -> int:
stack = []
sign = '+'
num = 0
while len(s) > 0:
c = s.pop(0)
if c.isdigit():
num = 10 * num + int(c)
# 遇到左括號開始遞歸計算 num
if c == '(':
num = helper(s)
if (not c.isdigit() and c != ' ') or len(s) == 0:
if sign == '+': ...
elif sign == '-': ...
elif sign == '*': ...
elif sign == '/': ...
num = 0
sign = c
# 遇到右括號返回遞歸結果
if c == ')': break
return sum(stack)
return helper(list(s))
你看,加了兩三行代碼,就可以處理括號了,這就是遞歸的魅力。至此,計算器的全部功能就實現了,通過對問題的層層拆解化整為零,再回頭看,這個問題似乎也沒那么復雜嘛。
PS:我認真寫了 100 多篇原創,手把手刷 200 道力扣題目,全部發布在 labuladong的算法小抄,持續更新。建議收藏,按照我的文章順序刷題,掌握各種算法套路后投再入題海就如魚得水了。
五、最后總結
本文借實現計算器的問題,主要想表達的是一種處理復雜問題的思路。
我們首先從字符串轉數字這個簡單問題開始,進而處理只包含加減法的算式,進而處理包含加減乘除四則運算的算式,進而處理空格字符,進而處理包含括號的算式。
可見,對於一些比較困難的問題,其解法並不是一蹴而就的,而是步步推進,螺旋上升的。如果一開始給你原題,你不會做,甚至看不懂答案,都很正常,關鍵在於我們自己如何簡化問題,如何以退為進。
退而求其次是一種很聰明策略。你想想啊,假設這是一道考試題,你不會實現這個計算器,但是你寫了字符串轉整數的算法並指出了容易溢出的陷阱,那起碼可以得 20 分吧;如果你能夠處理加減法,那可以得 40 分吧;如果你能處理加減乘除四則運算,那起碼夠 70 分了;再加上處理空格字符,80 有了吧。我就是不會處理括號,那就算了,80 已經很 OK 了好不好。
_____________
我的 在線電子書 有 100 篇原創文章,手把手帶刷 200 道力扣題目,建議收藏!對應的 GitHub 算法倉庫 已經獲得了 70k star,歡迎標星!