[Leetcode]_150 Evaluate Reverse Polish Notation

/**
 *  Index: 150
 *  Title: Evaluate Reverse Polish Notation
 *  Author: ltree98
 **/

题意

计算逆波兰式(逆波兰式也可称作后缀表示法,它将操作符置于操作数后面,这种方式不需要用括号来标识操作符的优先级)的值,合法的操作符有 + - * / ,每个操作数为一个整数或表达式。

注意:

  • 除法运算,需要向下取整
  • 给定的表达式一定合法。即一定会有个最终值,不会有除0等不合法行为

我的

思路

用栈结构,进行计算;注意 - / 运算符时,弹出栈操作数的位置顺序。

时间复杂度:O(n)
空间复杂度:O(n)

实现

class Solution {
public:
    int evalRPN(vector<string>& tokens) {
        stack<int> answer;
        
        for(int i = 0; i < tokens.size(); i++)  {
            if(tokens[i][0] == '+') {
                int param1 = answer.top();
                answer.pop();
                int param2 = answer.top();
                answer.pop();
                answer.push(param1 + param2);
            }
            else if(tokens[i][0] == '-' && tokens[i].length() == 1) {
                int param1 = answer.top();
                answer.pop();
                int param2 = answer.top();
                answer.pop();
                answer.push(param2 - param1);
            }
            else if(tokens[i][0] == '*')    {
                int param1 = answer.top();
                answer.pop();
                int param2 = answer.top();
                answer.pop();
                answer.push(param1 * param2);
            }
            else if(tokens[i][0] == '/') {
                int param1 = answer.top();
                answer.pop();
                int param2 = answer.top();
                answer.pop();
                answer.push(param2 / param1);
            }    
            else    {
                int val = std::stoi(tokens[i]);   
                answer.push(val);
            }
        }
        
        return answer.top();
    }
};

进阶

思路

基本都是通过栈来做,非常标准的栈练习题;差别在于代码的简洁度不同。

比如用switch替换一堆if-elseif-else,或者用lambda。

实现

class Solution {
public:
    int evalRPN(vector<string>& tokens) {
        unordered_map<string, function<int (int, int) > > map = {
            { "+" , [] (int a, int b) { return a + b; } },
            { "-" , [] (int a, int b) { return a - b; } },
            { "*" , [] (int a, int b) { return a * b; } },
            { "/" , [] (int a, int b) { return a / b; } }
        };
        std::stack<int> stack;
        for (string& s : tokens) {
            if (!map.count(s)) {
                stack.push(stoi(s));
            } else {
                int op1 = stack.top();
                stack.pop();
                int op2 = stack.top();
                stack.pop();
                stack.push(map[s](op2, op1));
            }
        }
        return stack.top();
    }
};

author: hercule24

发布了472 篇原创文章 · 获赞 387 · 访问量 78万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览