题目主要信息

1、给定一个表达式求值

2、字符串中的有效字符包括[‘0’-‘9’],‘+’,‘-’, ‘*’,‘/’ ,‘(’, ‘)’,‘[’, ‘]’,‘{’ ,‘}’

3、字符串一定合法

4、可能出现多个数字连在一起组成多位数

5、可能出现负数

方法一:利用栈求解

具体方法

我们可以把加减乘除分成2组,分别是+-和*/

然后用o1和o2来储存当前符号,具体定义为

+,o1=1,-,o1=-1

*,o2=1,/,o2=-1

我们通过表达式res=num1+o1×num2来保存当前算式,其中num1为已经计算部分的值,而num2是正在计算部分的值,num1和num2中间由+-隔开,其中num2正在通过×/计算出结果

当遇到数字时,我们直接获取数字,并且计算o2,获得num2

当遇到+-时,代表前面的公式已经计算完毕,计算num1,并更新o1,o2,num2

当遇到×/时,需要对num2进行更新

当遇到左括号时,我们要把现在所有结果入栈,并且重置参数

当遇到右括号时,我们要获得当前数字来更新num2

Java代码

import java.util.Arrays;
import java.util.Scanner;

import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String s = sc.nextLine();
        int n = s.length();
        int num1 = 0;
        int o1 = 1;
        int num2 = 1;
        int o2 = 1;
        Stack<Integer> stk = new Stack<>();

        for(int i=0; i<n; i++){
            char c = s.charAt(i);
            if(Character.isDigit(c)){  //遇到数字则定义num2
                int cur = 0;
                while(i<n && Character.isDigit(s.charAt(i))){
                    cur = cur * 10 + (s.charAt(i) - '0');
                    i++;
                }
                i--;
                num2 = o2 == 1 ? num2 * cur : num2 / cur;
            }else if(c=='*' || c=='/'){  //遇到×÷定义o2
                o2 = c == '*' ? 1 : -1;
            }else if(c == '(' || c=='{' || c=='['){  //遇到括号则保存当前结果,并初始化
                stk.push(num1);
                stk.push(o1);
                stk.push(num2);
                stk.push(o2);

                num1 = 0;
                o1 = 1;
                num2 = 1;
                o2 = 1;
            }else if(c == '+' || c=='-'){  //遇到加减,说明可以开始计算,计算num1并对定义其他几个变量
                if(c=='-' && (i==0 || s.charAt(i-1)=='(' || s.charAt(i-1)=='[' || s.charAt(i-1)=='{')){
                    o1 = -1;
                    continue;
                }
                num1 = num1 + o1 * num2;
                o1 = (c == '+' ? 1 : -1);
                num2 = 1;
                o2 = 1;
            }else{  //遇到右括号,则出栈,并计算num2
                int cur = num1 + o1 * num2;
                o2 = stk.pop();
                num2 = stk.pop();
                o1 = stk.pop();
                num1 = stk.pop();

                num2 = o2 == 1 ? num2 * cur : num2 / cur;
            }
        }
        System.out.println(num1 + o1 * num2);
    }
}


复杂度分析

  • 时间复杂度:O(n)O(n),时间复杂度和表达式长度正相关
  • 空间复杂度:O(n)O(n),需要一个栈,长度和表达式长度正相关

方法二:逆波兰表达式

具体方法

这道题可以通过逆波兰表达式来进行计算。 逆波兰计算器的原理是使用逆波兰表达式来计算出表达式的值,我们人类能够熟练使用的是中缀表达式,比如2×(9+6/3-5)+4就是一个中缀表达式,但是看到上面的简单计算器就知道处理起来很麻烦。于是有一种逆波兰计算器,计算是在逆波兰表达式(也叫做后缀表达式)的基础上。   逆波兰计算器的计算过程为:从左到右扫描后缀表达式,遇到数字就入栈,遇到操作符就从栈弹出两个数字,然后计算得到的值继续入栈,继续扫描表达式,直到扫描完毕得到结果。 alt 从逆波兰计算器的扫描过程可以看到,过程特别简单,代码写起来也比较容易。但现在的难点在于:如何把中缀表达式转成后缀表达式?

  这个过程已经有大神给出来了,我们记下来即可:

  1、初始化两个栈:运算符栈s1和存储中间结果的栈s2;

  2、从左到右扫描中缀表达式;

  3、遇到操作数时,压入到栈s2;

  4、遇到运算符时:

    1)如果s1为空或s1栈顶为左括号"(",则压入到s1;

    2)不满足1),则和s1栈顶运算符比较优先级,高于,则压入s1;

    3)不满足1)和2),弹出s1栈顶运算符并压入到s2,再次回到2)。

  5、遇到右括号")“时,依此弹出s1并压入s2,直到遇到左括号”)"为止,此时丢掉一对括号;

  6、重复2-5,直到扫描完毕;

  7、将s2栈弹出压入到s1,然后s1弹出全部,弹出的顺序即为后缀表达式。

Java代码

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String s = sc.nextLine();

        List<String> infix = expressionToList(s);  // List
        List<String> suffix = infixToSuffix(infix); // 中缀转后缀
        Stack<String> stk = new Stack<>();    // 存储中间结果
        // 逆波兰计算器
        for (int i = 0; i < suffix.size(); i++) {
            String tmp = suffix.get(i);
            if (isOper(tmp)) {
                String num2 = stk.pop();
                String num1 = stk.pop();
                String reuslt = cal(num1, tmp, num2);
                stk.push(reuslt);
            } else { // 数字直接入栈
                stk.push(tmp);
            }
        }
        System.out.println(stk.pop());
    }

    public static String cal(String num1, String oper, String num2) {
        Long result = 0l;
        Long a = Long.parseLong(num1);
        Long b = Long.parseLong(num2);
        switch (oper) {
            case "+":
                result = a + b;
                break;
            case "-":
                result = a - b;
                break;
            case "*":
                result = a * b;
                break;
            case "/":
                result = a / b;
                break;
        }
        return String.valueOf(result);
    }

    public static List<String> infixToSuffix(List<String> infix) {
        List<String> suffix = new ArrayList<>();
        Stack<String> stack1 = new Stack<>();   // 只用于保存操作符
        Stack<String> stack2 = new Stack<>();   // 用于保存中间结果
        for (int i = 0; i < infix.size(); i++) {
            String tmp = infix.get(i);
            if (isOper(tmp)) {   // 操作符要根据情况来入栈 1
                if (stack1.isEmpty() || "(".equals(tmp) || "[".equals(tmp) || "{".equals(tmp)) {
                    // 如果 stack1 是空的,或者 tmp 是左括号,直接入栈
                    stack1.push(tmp);
                } else { // stack1 不是空的,且 tmp 不是左括号
                    if (")".equals(tmp) || "]".equals(tmp) || "}".equals(tmp)) {
                        // tmp 是右括号,则把 stack1 遇到左括号之前,全部倒入 stack2
                        while (!("(".equals(stack1.peek()) || "[".equals(stack1.peek()) || "{".equals(stack1.peek()))) {
                            stack2.push(stack1.pop());
                        }
                        stack1.pop();   // 丢掉左括号
                    } else {
                        // tmp 是 +-*/ 其中之一
                        while (!stack1.isEmpty() && priority(stack1.peek()) >= priority(tmp)) {
                            // 在 tmp 能够碾压 stack1 的栈顶操作符之前,把 stack1 的栈顶操作符倒入 stack 2 中
                            stack2.push(stack1.pop());
                        }
                        // 离开 while 时,要么 stack1 已经倒空了,要么就是现在 tmp 可以压住 stack.peek() 了
                        stack1.push(tmp);
                    }
                }
            } else { // 操作数直接入栈 2
                stack2.push(tmp);
            }
        }

        // stack1 剩余操作符全部倒入 stack2
        while (!stack1.isEmpty()) {
            stack2.push(stack1.pop());
        }

        // stack2 的逆序才是结果,所以要再倒一次
        while (!stack2.isEmpty()) {
            stack1.push(stack2.pop());
        }

        // 现在 stack1 的元素倒出来的顺序就是后缀表达式
        while (!stack1.isEmpty()) {
            suffix.add(stack1.pop());
        }

        return suffix;
    }

    public static List<String> expressionToList(String expression) {
        List<String> list = new ArrayList<>();
        int len = expression.length();
        String keepNum = "";
        for (int i = 0; i < len; i++) {
            char c = expression.charAt(i);
            if (Character.isDigit(c)) {
                if (i != len - 1 && Character.isDigit(expression.charAt(i + 1))) {
                    // 如果下一个字符也是数字
                    keepNum += c;
                } else {
                    // 当前是最后一个字符,或者下一个开始不是数字
                    keepNum += c;
                    list.add(keepNum);
                    keepNum = "";
                }
            }else if(c == '-'){
                if(i==0 || expression.charAt(i-1)=='(' || expression.charAt(i-1)=='[' || expression.charAt(i-1)=='{'){
                    keepNum += c;
                }else{
                    list.add(c + "");
                }
            }else{
                list.add(c + "");
            }
        }
        return list;
    }

    public static boolean isOper(String str) {
        return "+".equals(str) || "-".equals(str) || "*".equals(str) || "/".equals(str) ||
                "(".equals(str) || ")".equals(str) || "[".equals(str) || "]".equals(str) ||
                "{".equals(str) || "}".equals(str);
    }

    public static int priority(String oper) {
        if ("+".equals(oper) || "-".equals(oper)) {
            return 0;
        } else if ("*".equals(oper) || "/".equals(oper)) {
            return 1;
        } else {
            return -1;
        }
    }

}

复杂度分析

  • 时间复杂度:O(n)O(n),时间复杂度和表达式长度正相关
  • 空间复杂度:O(n)O(n),需要一个栈,长度和表达式长度正相关