[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

Solution:

需要注意operand为负数的情况。

class Solution {

public:

    int stringToInt(string s)

    {

        if(s.length() == 0)

            return 0;

        int p = 1;

        if(s[0] == '-') 

        {

            p = -1;

            s = s.substr(1, s.length() - 1);

        }

        int ans = s[0] - '0';

        for(int i = 1;i < s.length();i++)

            ans = ans * 10 + s[i] - '0';

        return ans * p;

    }

    

    int evalRPN(vector<string> &tokens) {

        int len = tokens.size(), ans = 0;

        int *operand = new int[len + 1], operandNum = 0;

        memset(operand, 0, (len + 1) * sizeof(int));

        

        int res = 0; // 0 for num, 1 +, 2 -, 3 *, 4 /

        for(int i = 0;i < len;i++)

        {

            if(tokens[i] == "+")

            {

                operand[operandNum - 2] = operand[operandNum - 2] + operand[operandNum - 1];

                operandNum--;

            }

            else if(tokens[i] == "-")

            {

                operand[operandNum - 2] = operand[operandNum - 2] - operand[operandNum - 1];

                operandNum--;

            }

            else if(tokens[i] == "*")

            {

                operand[operandNum - 2] = operand[operandNum - 2] * operand[operandNum - 1];

                operandNum--;

            }

            else if(tokens[i] == "/")

            {

                operand[operandNum - 2] = operand[operandNum - 2] / operand[operandNum - 1];

                operandNum--;

            }

            else

            {

                int op = stringToInt(tokens[i]);

                operand[operandNum++] = op;

            }

        }

        return operand[0];

    }

};

 

你可能感兴趣的:(LeetCode)