1. 程式人生 > >【LeetCode】Evaluate Reverse Polish Notation 解題報告

【LeetCode】Evaluate Reverse Polish Notation 解題報告

【題意】

計算逆波蘭表示式(又叫字尾表示式)的值。

例如:

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

【思路】

用一個儲存運算元,遇到運算元直接壓入棧內,遇到操作符就把棧頂的兩個運算元拿出來運算一下,然後把運算結果放入棧內。

【Java程式碼】
public class Solution {
    public int evalRPN(String[] tokens) {
        int ret = 0;
        Stack<Integer> num = new Stack<Integer>();
        for (int i = 0; i < tokens.length; i++) {
        	if (isOperator(tokens[i])) {
        		int b = num.pop();
        		int a = num.pop();
        		num.push(calc(a, b, tokens[i]));
        	} else {
        		num.push(Integer.valueOf(tokens[i]));
        	}
        }
        ret = num.pop();
        return ret;
    }
	
	boolean isOperator(String str) {
		if (str.equals("+") || str.equals("-") || str.equals("*") || str.equals("/")) 
			return true;
		return false;
	}
	
	int calc(int a, int b, String operator) {
		char op = operator.charAt(0);
		switch (op) {
		case '+': return a + b;
		case '-': return a - b;
		case '*': return a * b;
		case '/': return a / b;
		}
		return 0;
	}
}