[LeetCode]227. Basic Calculator II

一、题目

Implement a basic calculator to evaluate a simple expression string.

The expression string contains only non-negative integers, +, -, *, / operators and empty spaces . The integer division should truncate toward zero.

You may assume that the given expression is always valid.

Some examples:

"3+2*2" = 7
" 3/2 " = 1
" 3+5 / 2 " = 5

二、思路

参考点击

对于每一个数，都默认一个符号，比如：3+5/2，3的符号是+，5的符号是+，2的符号是/。

符号优先级不同：乘除马上操作，得到结果保存；而加减不动。读取字符串，如果当前字符是数字，那么临时保存（注意数字可能是多位的，故需要num=num*10+s.charAt(i)-'0'）。如果当前字符是+，则num入栈；如果当前字符是-，则num取负数入栈；如果当前字符是*，则取栈顶与该num相乘，入栈；如果当前字符是/，则取栈顶与num相除然后入栈。最后，对栈中元素求和，为最终结果。

三、代码实现

  public int calculate(String s) {
        int re = 0;
        Stack<Integer> st = new Stack<>();
        char sign = '+';
        int num = 0;
        for(int i=0;i<s.length();i++){
            char cur = s.charAt(i);
            if(cur>='0'&&cur<='9'){
                num = num*10+(cur-'0');
            }
            if((cur!=' '&&(cur<'0'||cur>'9'))||i==s.length()-1){
                if(sign=='-')
                    st.push(-num);
                else if(sign=='+')
                    st.push(num);
                else if(sign=='*')
                    st.push(st.pop()*num);
                else if(sign=='/')
                    st.push(st.pop()/num);
                sign=cur;
                num=0;
            }
        }
        while(!st.isEmpty()){
            re += st.pop();
        }
        return re;
    }

四、错误总结

1、边界情况：只有单个数字，比如s=‘3’，仍需要入栈，故加了i==s.length()-1

2、很trick的一点是，发现当前字符是符号，但是去比较私密符号的时候，用的是上一次的符号；即当前字符如果是符号，说明要操作该符号之前的数字，而不是其之后的数字，体现在代码中的是用sign去比较，而不是用cur比较。