LeetCode0241.为运算表达式设计优先级

0241.为运算表达式设计优先级

描述

给定一个含有数字和运算符的字符串，为表达式添加括号，改变其运算优先级以求出不同的结果。你需要给出所有可能的组合的结果。有效的运算符号包含 +, - 以及 * 。

实例

输入: "2-1-1"
输出: [0, 2]
解释: 
((2-1)-1) = 0 
(2-(1-1)) = 2

输入: "2*3-4*5"
输出: [-34, -14, -10, -10, 10]
解释: 
(2*(3-(4*5))) = -34 
((2*3)-(4*5)) = -14 
((2*(3-4))*5) = -10 
(2*((3-4)*5)) = -10 
(((2*3)-4)*5) = 10

题解

• 对于每一个运算符,都可以作为最后一割运算,对左右两边的结果进行操作
• 要获得左右两边的结果,都可以视作一个新的输入
• 如果某一个字符串中没有操作符,说明是单个数字,需要直接存入res

    public static List<Integer> diffWaysToCompute(String input) {
        List<Integer> res = new ArrayList<>();

        for (int i = 0; i < input.length(); i++) {
            char c = input.charAt(i);
            if (c == '*' || c == '+' || c == '-'){
                List<Integer> resLeft = diffWaysToCompute(input.substring(0,i));
                List<Integer> resRight = diffWaysToCompute(input.substring(i+1,input.length()));

                if (c == '*'){
                    for (int intLeft:resLeft)
                        for (int intRight:resRight)
                            res.add(intLeft * intRight);
                } else if (c == '+'){
                    for (int intLeft:resLeft)
                        for (int intRight:resRight)
                            res.add(intLeft + intRight);
                } else if (c == '-'){
                    for (int intLeft:resLeft)
                        for (int intRight:resRight)
                            res.add(intLeft - intRight);
                }
            }
        }

        if (res.isEmpty()){
            res.add(Integer.parseInt(input));
        }

        return res;
    }