150.逆波兰表达式

原题链接:150.逆波兰表达式

思路:
逆波兰表达式其实就是后缀表达式
遍历字符串,先判断当前字符是否是运算符号,如果是 则取两次栈顶的元素,赋值给两个变量
然后拿这两个变量和运算符号进行运算 运算完毕后压入栈中等待下次运算
如果当前字符不是运算符号,则将数字压入栈中等待运算即可

最后把栈顶元素取出,这个就是题目需要的值

全代码:

class Solution {
public:
    int evalRPN(vector<string>& tokens) {
        stack<long long> st; 
        for (int i = 0; i < tokens.size(); i++) {
            if (tokens[i] == "+" || tokens[i] == "-" || tokens[i] == "*" || tokens[i] == "/") {
                long long num1 = st.top();
                st.pop();
                long long num2 = st.top();
                st.pop();
                if (tokens[i] == "+") st.push(num2 + num1);
                if (tokens[i] == "-") st.push(num2 - num1);
                if (tokens[i] == "*") st.push(num2 * num1);
                if (tokens[i] == "/") st.push(num2 / num1);
            } else {
                st.push(stoll(tokens[i]));
            }
        }

        int result = st.top();
        st.pop(); // 把栈里最后一个元素弹出(其实不弹出也没事)
        return result;
    }
};

你可能感兴趣的:(数据结构,leetcode)