力扣题目 分治及递归 思路

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

• 每个运算符都可以当做一个分隔符将字符串分为左右两个子字符串，左右两个子字符串中如果也有运算符，则可以继续分割，直到字符串分割的只有数字为止
• 每次分割相当于加括号（左右子字符串各加一个括号）
• zhao1()函数的作用是找出当前字符串的所有可能结果，放在当前的res列表中，层层递归，到最底层后，然后返回。最终返回给顶层，进行最外层的运算符运算得出最终结果放在顶层res列表里，这个res列表中的结果即最终结果

def zhao1(s):
    # 每次递归使用这个函数时，都会产生一个结果列表
    res = []
    if s.isdigit():
        return [int(s)]
    for i, j in enumerate(s):
        if j in ['-', '+', '*']:
            for l in zhao1(s[:i]):
                for r in zhao1(s[i+1:]):
                    res.append(eval(f'{l}{j}{r}'))
                    # if j == '+':
                    #     res.append(l + r)
                    # elif j == '-':
                    #     res.append(l - r)
                    # else:
                    #     res.append(l * r)
    return res

print(zhao1('2*3-4*5'))
print(zhao1('2-1-1'))
print(zhao1('11'))
print(zhao1('10+5'))

结果

[-34, -10, -14, -10, 10]
[2, 0]
[11]
[15]

附：
最开始我看到这个题是例子2输入是23-45
中间也没有空格，我就想怎么变成输入 2*3-4*5
也相当于一道小练习吧，本来想用 insert() 方法但好像不行，这里写下我的方法

s = '23-45'
s = [i for i in s]
for i in range(1, len(s)):
	if s[i].isdigit() and s[i-1].isdigit():
		s[i] = '*' + s[i]
s = ''.join(s)