[leetCode]241. 为运算表达式设计优先级

leetCode

链接：https://leetcode-cn.com/problems/different-ways-to-add-parentheses

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

示例 1:

输入: "2-1-1"
输出: [0, 2]
解释: 
((2-1)-1) = 0 
(2-(1-1)) = 2
示例 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

分治

思路：
加括号就相当于把表达式"2-1-1"分成两部分，有两种形式:

• (2-1)-1-----> 2-1 与 1
• 2-(1-1) -----> 2 与 1 - 1

通过观察可以发现可以通过运算符将表达式分割成左右两部分，不断分割直到分割到只剩下数字。将分割到最后的结果加入集合中，得到左右两个集合，从两个集合中取结果进行组合然后根据当前的运算符进行运算，将结果加入集合中。

class Solution {
    public List<Integer> diffWaysToCompute(String input) {
        List<Integer> ans = new ArrayList<>();
        for (int i = 0; i < input.length(); i++) {
            char op = input.charAt(i);
            if (op == '+' || op == '-' || op == '*') {
                String left = input.substring(0, i);
                String right = input.substring(i + 1);
                List<Integer> aLeft = diffWaysToCompute(left);
                List<Integer> aRight = diffWaysToCompute(right); 
                for (Integer l : aLeft) {
                    for (Integer r : aRight) {
                        if (op == '+') {
                            ans.add(l+r);
                        } else if (op == '-') {
                            ans.add(l-r);
                        } else {
                            ans.add(l*r);
                        }
                    }
                }
            }
        }
        if (ans.size() == 0) {
            ans.add(Integer.valueOf(input));
        }
        return ans;
    }
}

优化

由于在分割过程中会出现重复的子表达式，这时候可以将该表达式对应的组合结果集合构成一个映射，当需要再次计算时直接返回对应的结果集，而不用再次计算

class Solution {

    Map<String, List<Integer>> map = new HashMap<>();

    public List<Integer> diffWaysToCompute(String input) {
        List<Integer> ans = new ArrayList<>();
        if (map.containsKey(input))
            return map.get(input);
        for (int i = 0; i < input.length(); i++) {
            char op = input.charAt(i);
            if (op == '+' || op == '-' || op == '*') {
                String left = input.substring(0, i);
                String right = input.substring(i + 1);
                List<Integer> aLeft = diffWaysToCompute(left);
                List<Integer> aRight = diffWaysToCompute(right); 
                for (Integer l : aLeft) {
                    for (Integer r : aRight) {
                        if (op == '+') {
                            ans.add(l+r);
                        } else if (op == '-') {
                            ans.add(l-r);
                        } else {
                            ans.add(l*r);
                        }
                    }
                }
            }
        }
        if (ans.size() == 0) {
            ans.add(Integer.valueOf(input));
        }
        map.put(input, ans);
        return ans;
    }
}