算法通关村第四关——计算器问题

设计算法实现计算机功能

输入:s = "3+2*2"
输出:7

使用栈可以很好的解决运算器的问题。由于乘除优先于加减计算,因此不妨考虑先进行所有乘除运算,并将这些乘除运算后的整数值放回原表达式的相应位置,则随后整个表达式的值,就等于一系列整数加减后的值。

public int calculate(String s) {
        Deque stack = new ArrayDeque();
        char preSign = '+';
        int num = 0;
        int n = s.length();
        for (int i = 0; i < n; ++i) {
            if (Character.isDigit(s.charAt(i))) {
                // 这里*10是为了现在要对一个两位数进行处理,前一层循环没有对num=0
                num = num * 10 + s.charAt(i) - '0';
            }
            if (!Character.isDigit(s.charAt(i)) && s.charAt(i) != ' ' || i == n - 1) {
                switch (preSign) {
                    case '+':
                        stack.push(num);
                        break;
                    case '-':
                        stack.push(-num);
                        break;
                    case '*':
                        stack.push(stack.pop() * num);
                        break;
                    default:
                        stack.push(stack.pop() / num);
                }
                preSign = s.charAt(i);
                num = 0;
            }
        }
        int ans = 0;
        while (!stack.isEmpty()) {
            ans += stack.pop();
        }
        return ans;
    }

就拿上面的示例来说,进入循环,3是数字,num = 3,第二个if不满足,

第二层循环,s.charAt(i)为+号,进入第二个if,进入第一个case,向栈中压入3,preSign=+,num=0,

第三层循环,s.charAt(i)为2,进入第一个if,num=2

第四层循环,s.charAt(i)为*号,但是现在的preSign为+号,进入第二个if,进入第一个case,向栈中压入2,preSign = *,num = 0

第五层循环,s.charAt(i)为2,进入第一个if,num=2,在第二个if的判断中,满足i == n - 1,进入第三个case,向栈中压入栈顶元素与现在num相乘的值,但是这里的操作是pop栈顶元素,然后与num相乘之后再次压入,现在栈中又两个数,分别是3和4。preSign = 2,num = 0

你可能感兴趣的:(算法,java,数据结构)