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

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

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

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

第一种思路：

这是一个典型的大问题可以被铲分为相似的小问题 的问题，所以可以用递归解题，

举例：要计算 1*3 - 2 + 4，

可以被拆成 diffWaysToCompute(1) * diffWaysToCompute(3 - 2 + 4)，

也可以被拆成 diffWaysToCompute(1 * 3) - diffWaysToCompute(2 + 4）

计算出子问题的解之后再根据中间的运算符组合一下：

比如假设左半部分得到的解为[2, 3]，右半部分得到的解为[1,0]，中间的运算符为 加号，

那么组合起来就可以得到[2 + 1, 2+ 0, 3 + 1, 3 + 0]的整体解。

 

class Solution(object):
    def diffWaysToCompute(self, input):
        """
        :type input: str
        :rtype: List[int]
        """
        if input.isdigit():
            return [int(input)]
        res = []
        
        for i in range(len(input)):
            if input[i] in "+-*":
                left = self.diffWaysToCompute(input[:i])
                right = self.diffWaysToCompute(input[i + 1:])
                
                for l in left:
                    for r in right:
                        if input[i] == "+":
                            res.append(l + r)
                        elif input[i] == "-":
                            res.append(l - r)
                        else:
                            res.append(l * r)

        return res

下面的用了eval，会更简洁 

class Solution(object):
    def diffWaysToCompute(self, input):
        """
        :type input: str
        :rtype: List[int]
        """
        if input.isdigit(): #如果没有运算
            return [int(input)] #那么直接返回数字
        
        res = []
        for i, char in enumerate(input):
            if char in ["+", "-", "*"]:
                left = self.diffWaysToCompute(input[:i]) #计算前半截表达式的值
                right = self.diffWaysToCompute(input[i + 1:]) #计算后半截表达式的值
                
                for l in left:
                    for r in right:
                        res.append(eval(str(l) + char + str(r))) #组合起来

        return res

第二种思路：

考虑到同一个表达式可能会进行多次重复的计算，所以可以使用字典记录已知问题的解，

比如假设我们已经计算过了 1+ 2的值是3， 那么下一次将不再运算，而是直接从字典里读取结果。

class Solution(object):
    def diffWaysToCompute(self, input):
        """
        :type input: str
        :rtype: List[int]
        """
        self.record = collections.defaultdict(list)
        self.work(input)
        return self.record[input]
        
        
        
    def work(self, input):
        if not input:
            return []
        if input in self.record:
            return self.record[input]
        
        if input.isdigit():
            self.record[input] = [int(input)]
            return self.record[input]
        res = []
        
        for i in range(len(input)):
            if input[i] in "+-*":
                left = self.diffWaysToCompute(input[:i])
                right = self.diffWaysToCompute(input[i + 1:])
                
                for l in left:
                    for r in right:
                        if input[i] == "+":
                            res.append(l + r)
                        elif input[i] == "-":
                            res.append(l - r)
                        else:
                            res.append(l * r)
                            
        self.record[input] = res
        return self.record[input]