描述
请写一个整数计算器,支持加减乘三种运算和括号。
示例
输入:
"1+2"
返回值:
3
思路:
题目给出了一个中缀表达式,要我们求表达式的结果,最直接的想法就是把中缀表达式转为后缀表达式,再利用一个栈完成后缀表达式的求值。
方法一:中缀表达式转后缀表达式+后缀表达式求值
中缀表达式转后缀表达式:
- 定义一个保存运算符的优先级,乘除优先级高于加减
- 定义一个动态数组保存后缀表达式,运算符栈存储运算符
- 遇到左括号时直接入运算符栈,遇到右括号时不断弹出栈中的运算符,并将其加入后缀表达式直到左括号出栈,左括号不加入后缀表达式
- 遇到操作数时,如果是多位数则需要将连续的多个数字取出,使用u保存整个数字字符串,并将数字字符串加入后缀表达式
后缀表达式求值:
先定义一个栈,遍历后缀表达式数组,遇到数字则将其转换为整型并入栈,遇到运算符则弹出栈顶元素,先弹出的数字为右操作数,后弹出的数字为左操作数,将计算结果入栈,直到遍历完后缀表达式,此时栈中的元素即为后缀表达式的结果,将计算结果出栈。
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * 返回表达式的值 * @param s string字符串 待计算的表达式 * @return int整型 */ Map<Character,Integer>map=new HashMap<Character,Integer>(); public int solve (String s) { // write code here //去除所有空格 s=s.replaceAll(" ",""); //用一个hashmap保存运算符的优先级 map.put('+',1); map.put('-',1); map.put('*',2); map.put('/',2); //保存后缀表达式的数组 ArrayList<String>list=new ArrayList(); //保存运算符的栈 Stack<Character>operation=new Stack<>(); int i; for(i=0;i<s.length();i++){ Character c=s.charAt(i); //计算操作数的值 if(isNumber(c)){ String u=""; int j=i; //从i位置开始后面的连续数字整体取出,加入nums while(j<s.length()&&isNumber(s.charAt(j)))u+=Character.toString(s.charAt(j++)); list.add(u); //更新i的值,因为i会自增,所以这里i=j-1 i=j-1; } //左括号直接入栈 else if(c=='(')operation.push(c); //遇到右括号,弹出栈中的运算符直到遇到左括号,运算符加入后缀表达式,左括号也需要出栈,但是左括号不需加入后缀表达式 else if(c==')') { Character c2; while((c2=operation.pop())!='(') { list.add(c2.toString()); } } else { //遇到运算符时,栈非空而且栈顶运算符优先级高于当前运算符,将运算符出栈并加入后缀表达式 //当栈顶元素为左括号时,条件不成立,当前运算符应该直接入栈 while(!operation.isEmpty()&&map.containsKey(operation.peek())&&map.get(operation.peek())>=map.get(c)) list.add(operation.pop().toString()); //将当前运算符入栈 operation.push(c); } } //将栈中的剩余运算符出栈并且加入后缀表达式中 while(!operation.isEmpty())list.add(operation.pop().toString()); Stack<Integer>number=new Stack<>(); for(String str:list){ if(!(str.equals("+")||str.equals("-")||str.equals("*")||str.equals("/")))number.push(Integer.valueOf(str)); else{ //先出栈的数字为右操作数,后出栈的为左操作数 int num2=number.pop(); int num1=number.pop(); switch(str){ //每次将运算结果压入栈 case "+":number.push(num1+num2);break; case "-":number.push(num1-num2);break; case "*":number.push(num1*num2);break; case "/":number.push(num1/num2);break; default: } } } return number.pop(); } //判定是否是数字 boolean isNumber(char c) { return Character.isDigit(c); } }
复杂度:
时间复杂度:,其中为字符串的长度,需要遍历字符串一次,计算表达式的值
空间复杂度:,空间复杂度主要取决于栈的空间和后缀表达式数组的大小,栈的元素个数和数组元素个数都不超过
方法二:栈+递归
相比于方法一,方法二代码会比较精简
1.先去除字符串中的空格
2.定义一个栈保存各部分表达式的和
3.遍历表达式,使用 变量记录运算符,使用 变量记录字符串中的数字部分的数字值
- 遇到数字时继续遍历求这个完整的数字的值,保存到 中
- 遇到左括号时先找到对应的右括号,因为里面可能嵌有多对括号,使用一个变量 记录括号对数直到变量为0,递归求这个括号里面的表达式的值,递归终止条件是求出最里面一层括号里表达式的值
- 遇到运算符时或者到表达式末尾时,就去计算上一个运算符并把计算结果 push进栈,然后保存新的运算符到
- 如果是 + ,直接push 进去
- 如果是 - ,push 进去负的当前数
- 如果是 ×、÷ ,pop 出一个运算数和当前数作计算
4.最后把栈中的结果求和即可
具体代码如下:
import java.util.*; public class Solution { //判定是否是数字 boolean isNumber(char c) { return Character.isDigit(c); } public int solve(String s){ //去除字符串中的空格 s=s.trim(); //定义一个栈保存各部分计算出来的和 Deque<Integer>stack=new ArrayDeque<>(); char[]charArr=s.toCharArray(); int num=0; char op='+'; int n=charArr.length; for(int i=0;i<n;i++){ char c=charArr[i]; if(isNumber(c)){ num=num*10+c-'0'; } //递归求出括号里面表达式的值 if(c=='('){ int j=i+1; //brackets记录括号数 int brackets=1; while(brackets>0){ if(charArr[j]=='('){ brackets++; } if(charArr[j]==')'){ brackets--; } j++; } num=solve(s.substring(i+1,j-1)); i=j-1; } if(!isNumber(c)||i==n-1){ if(op=='+')stack.push(num); else if(op=='-')stack.push(-1*num); else if(op=='*')stack.push(stack.pop()*num); else if(op=='/')stack.push(stack.pop()/num); num=0; op=c; } } int res=0; //栈中结果求和 while(!stack.isEmpty()){ res+=stack.pop(); } return res; } }
复杂度:
时间复杂度:,为字符串的长度,至少遍历一次字符串
空间复杂度:,递归深度最大为字符串长度,每次递归空间复杂度为,因此递归空间复杂度为,存储各部分计算和的栈最大不超过,所以总的空间复杂度为.