LeetCode #241 为运算表达式设计优先级分治

最新推荐文章于 2022-09-08 22:01:05 发布

阿玮d博客

最新推荐文章于 2022-09-08 22:01:05 发布

阅读量287

点赞数 1

分类专栏： LeetCode # 分治文章标签：字符串 leetcode python

本文链接：https://blog.csdn.net/weixin_42511320/article/details/105263569

版权

LeetCode 同时被 2 个专栏收录

48 篇文章 0 订阅

订阅专栏

分治

2 篇文章 0 订阅

订阅专栏

LeetCode #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

思路分析

对于一个形如x or y（op 为运算符， x 和 y 为运算数）的算式而言，它的结果组合取决于 x 和 y 的结果组合数，而 x 和 y 又可以写成形如 x op y 的算式

因此该问题的子问题就是 x op y 中的 x 和 y：以运算符分隔的左右两侧算式解

分治算法三步走：

分解：按运算符分成左右两部分，分别求解
解决：实现一个递归函数，输入算式，返回算式解
合并：根据运算符合并两部分的解，得出最终解

class Solution:
    def diffWaysToCompute(self, input: str) -> List[int]:
    	# 如果只有数字，直接返回
        if input.isdigit():
            return [int(input)]
            
        res = []
        for i in range(len(input)):
            char = input[i]
            if char in ['+', '-', '*']:
                left = self.diffWaysToCompute(input[:i])
                right = self.diffWaysToCompute(input[i+1:])
                for l in left:
                    for r in right:
                        if char == '+':
                            res.append(l + r)
                        elif char == '-':
                            res.append(l - r)
                        else:
                            res.append(l * r)
        return res

时间复杂度： $O (N l o g N)$
空间复杂度：不会算。。。

备忘录优化

在递归算法中加入备忘录，相当于对递归剪枝，加快速度

class Solution:
    def diffWaysToCompute(self, input: str):
        # 递归 + 备忘录
        # 将input定义为私有属性
        self.formula = input
        self.memo = {}
        return self._diffWaysToCompute(0, len(input))

    def _diffWaysToCompute(self, lo, hi):
        if self.formula[lo:hi].isdigit():
            return [int(self.formula[lo:hi])]
        if((lo, hi) in self.memo):
            return self.memo.get((lo, hi))
        ret = []
        for i, char in enumerate(self.formula[lo:hi]):
            if char in ['+', '-', '*']:
                leftResult = self._diffWaysToCompute(lo, i + lo)
                rightResult = self._diffWaysToCompute(lo + i + 1, hi)
                # eval() 用于执行一个字符串表达式，并返回表达式的值
                ret.extend([eval(str(i) + char + str(j)) for i in leftResult for j in rightResult])
                # 将结果加入备忘录
                self.memo[(lo, hi)] = ret
        return ret

作者：cui-jin-hao-_official
链接：https://leetcode-cn.com/problems/different-ways-to-add-parentheses/solution/python-fen-zhi-fa-di-gui-bei-wang-lu-by-cui-jin-ha/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。