【LeetCode】227. 基本计算器Ⅱ

代码

#include <stack>
using namespace std;

class Solution {
public:
    int calculate(string s) {
        int ans = 0;
        stack<int> st;
        stack<char> opst;
        int num = 0;
        //遍历字符串，将数字和符号入栈，当遇到符号且符号栈顶为乘除符号时，弹出元素将计算后的重新入栈
        for (int i = 0; i < s.length(); ++i) {
            char c = s[i];
            if (c == ' ')
                continue;
            if (isdigit(c)) { // 使用isdigit检查是否为数字
                num = num * 10 + (c - '0');
            } else {
                if (!opst.empty() && (opst.top() == '*' || opst.top() == '/')) {
                    num = (opst.top() == '*') ? st.top() * num : st.top() / num;
                    opst.pop();
                    st.pop();
                }
                st.push(num);
                num = 0;
                opst.push(c);
            }
        }
        // 处理遍历完字符串后，栈顶符号是乘除的情况
        if (!opst.empty() && (opst.top() == '*' || opst.top() == '/')) {
            num = (opst.top() == '*') ? st.top() * num : st.top() / num;
            opst.pop();
            st.pop();
        }
        st.push(num); 
        num = 0;
        //此时符号栈中应只有加减号
        while (!opst.empty()) {
            char op = opst.top();
            opst.pop();
            int prev = st.top();
            st.pop();
            ans = (op == '+') ? ans + prev : ans - prev;
        }
        ans += st.top();
        return ans;
    }
};