150-evaluate-reverse-polish-notation

Try it on leetcode

Description

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

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

Note that division between two integers should truncate toward zero.

It is guaranteed that the given RPN expression is always valid. That means the expression would always evaluate to a result, and there will not be any division by zero operation.

 

Example 1:

Input: tokens = ["2","1","+","3","*"]
Output: 9
Explanation: ((2 + 1) * 3) = 9

Example 2:

Input: tokens = ["4","13","5","/","+"]
Output: 6
Explanation: (4 + (13 / 5)) = 6

Example 3:

Input: tokens = ["10","6","9","3","+","-11","*","/","*","17","+","5","+"]
Output: 22
Explanation: ((10 * (6 / ((9 + 3) * -11))) + 17) + 5
= ((10 * (6 / (12 * -11))) + 17) + 5
= ((10 * (6 / -132)) + 17) + 5
= ((10 * 0) + 17) + 5
= (0 + 17) + 5
= 17 + 5
= 22

 

Constraints:

• 1 <= tokens.length <= 104
• tokens[i] is either an operator: "+", "-", "*", or "/", or an integer in the range [-200, 200].

Solution(Python)

class Solution:
    """
    refer https://en.wikipedia.org/wiki/Shunting-yard_algorithm

    iterate through the array if numbers are found push them to stack
    if an operator is found pop last two elements from stack and do that op

    last standing number is the vealuated result

    Time Complexity: O(n)
    Space Complexity: O(n)
    """

    def evalRPN(self, tokens: List[str]) -> int:
        s = []

        for token in tokens:
            if token in "+-*/":
                op1 = s.pop()
                op2 = s.pop()
                s.append(self.eval(op1, op2, token))
            else:
                s.append(int(token))
        return s[-1]

    def eval(self, op1, op2, op):
        print(op1, op2, op)
        if op == "+":
            return op1 + op2
        elif op == "-":
            return op2 - op1
        elif op == "*":
            return op1 * op2
        elif op == "/":
            return int(op2 / op1)