LeetCode -- Evaluate Reverse Polish Notation

题目如下:

Evaluate the value of an arithmetic expression in Reverse Polish Notation.

Valid operators are +-*/. Each operand may be an integer or another expression.

Some examples:

  ["2", "1", "+", "3", "*"] -> ((2 + 1) * 3) -> 9
  ["4", "13", "5", "/", "+"] -> (4 + (13 / 5)) -> 6
-----------------------------------------------------------------------------------

这个题目难度也不大,主要是利用栈后进先出的性质,每次遇到数字就进栈,遇到运算符就拿出两个数字用这个运算符计算,然后将计算的结果保存到栈中。

如果表达式没有错误的话,最后栈顶的元素就是计算的结果。

class Solution {
public:
    int evalRPN(vector<string> &tokens) {
        stack<int> stk;
        for(vector<string>::iterator it = tokens.begin(); it != tokens.end(); ++it)
        {
            if(isOper(*it) ){
                int b = stk.top(); stk.pop();
                int a = stk.top(); stk.pop();
                stk.push(rst(a, b, *it));
            }else{
                stk.push(numb(*it));
            }
        }
        return stk.top();
    }
    int numb(string s)
    {
        int ret = 0, flag = 1;
        size_t i = 0;
        if(s[i] == '-'){
            flag = -1;
            ++i;
        }else if(s[i] == '+'){
            ++i;
        }
        for(; i < s.length(); ++i)
        {
            ret = ret*10 + (s[i] - '0');
        }
        return ret*flag;
    }
    int rst(int a, int b, string op)
    {
        if(op == "+")
            return a+b;
        else if(op == "-")
            return a-b;
        else if(op == "*")
            return a*b;
        else if(op == "/")
            return a/b;
    }
    bool isOper(string s){
        return (s == "+" || s == "-" || s == "*" || s == "/");
    }
};
代码中,用isOper判断元素是运算符还是数字,如果是运算符就从栈中取两个数字,用rst来做计算,如果是数字的话就转化位数字直接进栈。

在进行数字转化的时候,考虑了数字符号为正和为负两种情况,不知道LeetCode的测试用例有没有考虑到这个情况。

代码的长度还是比较长的,可以进一步优化,不过这么写,更容易看懂吧


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值