Leetcode (10) 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

Reverse Polish Notation是多项式的一种表示方法，大概就是操作符保存在数字的后面，具体解释请出门左转见wiki（中文、英文）。

解法提示

根据逆波兰表示法的含义，我们很自然的可以想到使用栈去保存结果。遍历字符串数组，若为数字则压入栈，若为操作符，则弹出栈中上面两个数字，使用操作符进行运算，再将运算完的结果压入栈即可。

解题代码

这里默认操作符只有+、-、*、/四个，更复杂的操作符思想都是一样一样的。共写了三个函数，一个用于根据运算数和操作符进行计算（calculate），一个用于判断字符串是否为操作符（isOp），最后一个函数为计算的主要函数（evalRPN）。

class Solution {
public:
    int calculate(int n1, int n2, string sop)
    {
        if (sop.size() != 1)
        {
            cout << "wrong operator!" << endl;
            return 0;
        }
        char op = sop[0];
        int n;
        switch (op)
        {
        case '+': n = n1 + n2;
            break;
        case '-': n = n1 - n2;
            break;
        case '*': n = n1 * n2;
            break;
        case '/': n = n1 / n2;
            break;
        default:
            cout << "wrong operator!" << endl;
            return 0;
            break;
        }
        return n;
    }

    bool isOp(string s)
    {
        if (s.size() != 1)
            return false;

        char c = s[0];
        if (c == '+' || c == '-' || c == '*' || c == '/')
            return true;
        else
            return false;
    }
    int evalRPN(vector<string> &tokens) {
        int r = 0;
        stack<int> temp;
        for (size_t i = 0; i != tokens.size(); ++i)
        {
            if (!isOp(tokens[i]))
            {
                temp.push(stoi(tokens[i]));
            }
            else
            {
                int n2 = temp.top();
                temp.pop();
                int n1 = temp.top();
                temp.pop();
                r = calculate(n1, n2, tokens[i]);
                temp.push(r);
            }
        }

        if (temp.size() != 0)
        {
            return temp.top();
        }
        else{
            return 0;
        }
    }
};

后记

花了一下午时间终于把博客整完了，刷了一个星期的Leetcode在周末还是要整理一下复习复习的，从周一到现在共做了九道题目，原先只是打算每天抽一个小时做一道题目的，后几天却开始每天刷两道了。不惊感叹一下，积少成多，现在看到题目的思路也越来越快，也可以很快写出正确的代码了。后面在发现更好的解法之后也许也还会更新目前的Leetcode系列。