leetcode_241 为运算表达式设计优先级

150 阅读1分钟

要求

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

示例 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

核心代码

class Solution:
    def diffWaysToCompute(self, expression: str) -> List[int]:
        if expression.isdigit():
            return [int(expression)]
        res = []
        for i in range(len(expression)):
            if expression[i] in "-+*":
                res1 = self.diffWaysToCompute(expression[:i])
                res2 = self.diffWaysToCompute(expression[i + 1:])
                for j in res1:
                    for k in res2:
                        res.append(self.helper(j,k,expression[i]))
        return res
    
    def helper(self,m,n,op):
        if op == "+":
            return m + n
        elif op == "-":
            return m - n
        else:
            return m * n  

image.png

解题思路:分治法(divide and conquer:分而治之)2*3-4*5,首先先要遇到第一个*符号,分解成2、3-4*5 两个表达式,在向下细化。