1. 问题描述:
给定一个仅包含数字 0-9 的字符串和一个目标值,在数字之间添加二元运算符(不是一元)+、- 或 * ,返回所有能够得到目标值的表达式。
示例 1:
输入: num = "123", target = 6
输出: ["1+2+3", "1*2*3"]
示例 2:
输入: num = "232", target = 8
输出: ["2*3+2", "2+3*2"]
示例 3:
输入: num = "105", target = 5
输出: ["1*0+5","10-5"]
示例 4:
输入: num = "00", target = 0
输出: ["0+0", "0-0", "0*0"]
示例 5:
输入: num = "3456237490", target = 9191
输出: []
提示:
0 <= num.length <= 10
num
仅含数字
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/expression-add-operators
2. 思路分析:
分析题目可以知道我们可以使用递归搜索所有位置可能添加的操作符,最终使用栈来计算递归过程中存储的表达式但是这样计算表达式的结果会超时,在y中的视频中讲到可以使用代数结构对表达式进行优化,在递归的过程中维护表达式的值这样就不用在求解完表达式之后最后还要使用栈计算表达式的结果,这样我们在递归出口就可以直接判断表达式的值是否等于target即可,我们可以分类讨论下一个操作符的情况,其中()为当前枚举的num中的数字,可以分为三种情况:
感觉上面的代数结构是非常巧妙的对于任意的表达式都可以表示为上面的三种情况,因为我们考虑的是下一个操作符,所以当我们计算到num中的最后一个数字的时候,只需要考虑最后一个操作符为加号的情况即可,因为考虑为减号或者除号的运算结果都是一样的这样会造成答案重复了,也即最后一个数字只需要计算一次操作符即可。
3. 代码如下:
from typing import List
class Solution:
def dfs(self, num: str, target: int, a: int, b: int, index: int, path: str, res: List[str]):
if index == len(num):
if a == target:
res.append(path[0: len(path) - 1])
return
else:
c = 0
for i in range(index, len(num)):
c = c * 10 + int(num[i])
# 下一个操作符为+
self.dfs(num, target, a + b * c, 1, i + 1, path + str(c) + "+", res)
# 下一个操作符为-, 为了控制只搜索一次(上面搜索了一次+号的情况), 所以这里需要控制重复搜索, 加上一个i + 1 < len(num)的判断条件
if i + 1 < len(num):
self.dfs(num, target, a + b * c, -1, i + 1, path + str(c) + "-", res)
if i + 1 < len(num):
self.dfs(num, target, a, b * c, i + 1, path + str(c) + "*", res)
# 去除前导0
if num[index] == "0": break
def addOperators(self, num: str, target: int) -> List[str]:
res = list()
self.dfs(num, target, 0, 1, 0, "", res)
return res