设计算法实现计算机功能
输入:s = "3+2*2"
输出:7
使用栈可以很好的解决运算器的问题。由于乘除优先于加减计算,因此不妨考虑先进行所有乘除运算,并将这些乘除运算后的整数值放回原表达式的相应位置,则随后整个表达式的值,就等于一系列整数加减后的值。
public int calculate(String s) {
Deque<Integer> stack = new ArrayDeque<Integer>();
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