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