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

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

示例 1:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/different-ways-to-add-parentheses
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

class Solution {

    private List<Integer> solve(String expression, int start, int end) {
        if (start > end) {
            return Collections.emptyList();
        }

        int num = 0;
        int index = start;
        while (index <= end && Character.isDigit(expression.charAt(index))) {
            num = num * 10 + expression.charAt(index++) - '0';
        }

        if (index == end + 1) {
            return Collections.singletonList(num);
        }


        List<Integer> ans = new ArrayList<>();

        for (int i = start; i <= end; ++i) {
            if (!Character.isDigit(expression.charAt(i))) {
                List<Integer> left = solve(expression, start, i - 1);
                List<Integer> right = solve(expression, i + 1, end);

                for (int x : left) {
                    for (int y : right) {
                        switch (expression.charAt(i)) {
                            case '+':
                                ans.add(x + y);
                                break;
                            case '-':
                                ans.add(x - y);
                                break;
                            case '*':
                                ans.add(x * y);
                                break;
                            default:
                                break;
                        }
                    }
                }

            }
        }

        return ans;
    }


    public List<Integer> diffWaysToCompute(String expression) {
        if (expression == null || expression.length() == 0) {
            return Collections.emptyList();
        }
        return solve(expression, 0, expression.length() - 1);
    }
}
posted @ 2022-01-04 21:06  Tianyiya  阅读(39)  评论(0)    收藏  举报