【LeetCode】227. 基本计算器 II 结题报告 (C++)

原题地址：https://leetcode-cn.com/problems/basic-calculator-ii/

题目描述：

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

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

示例 1:

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

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

输入: " 3+5 / 2 "
输出: 5
说明：

你可以假设所给定的表达式都是有效的。
请不要使用内置的库函数 eval。
 

 

解题方案：

用个栈就完事了，其中计算的方式自己定义就行了，类似自动机。不过如何画自动机，已经忘光了。。

代码：

class Solution {
public:
    int calculate(string s) {
        stack<long> s1;
        stack<char> s2;
        long ans = 0;
        for(int i = 0; i < s.length(); i++){
            if(s[i] == ' ')    continue;       
            if(s[i] >= '0' && s[i] <= '9'){
                long tmp = s[i] - '0';
                while(s[i +1 ] >= '0' && s[i + 1] <= '9')
                    tmp = tmp * 10 + (s[++ i] - '0');
                ans = tmp;
                if(!s2.empty()){
                    if(s2.top() == '*' || s2.top() == '/'){
                        if(s2.top() == '*')    ans = s1.top() * ans;
                        else    ans = s1.top() / ans;
                        s1.pop();
                        s2.pop();
                    }
                }
                s1.push(ans);
            }
            else if(s[i] == '+' || s[i] == '-'){
                if(!s2.empty()){
                    char c = s2.top();
                    s2.pop();
                    ans = s1.top();
                    s1.pop();
                    if(c == '+')    ans = ans + s1.top();
                    else    ans = s1.top() - ans;
                    s1.pop();
                    s1.push(ans);
                }
                s2.push(s[i]);
            }
            else    
                s2.push(s[i]);
        }
        
        if(!s2.empty()){
            char c = s2.top();
            ans = s1.top();
            s1.pop();
            if(c == '+')    ans = ans + s1.top();
            else    ans = s1.top() - ans;
        }
        return ans;
    }
};