leetcode：241. 为运算表达式设计优先级【dfs + eval】

分析

每次选择其中的a ？ b进行加一对括号
知道用完所有运算符
然后放入一个set中防止运算式子相等即可

ac code

class Solution:
    def diffWaysToCompute(self, expression: str) -> List[int]:
        ans = []
        operators = ('+', '-', '*')
        def countOperator(s):
            cnt = 0
            for c in s:
                if c in operators:
                    cnt += 1
            
            return cnt
        
        s = set()
        def dfs(nums, ops):
            if len(nums) == 1:
                s.add(nums[0])
                return
            # 选一个进行计算
            for i in range(len(nums) - 1):
                new_num = '(' + nums[i] + ops[i] + nums[i + 1] + ')'

                new_nums = nums[:i] + [new_num] + nums[i + 2:]
                new_ops = ops[:i] + ops[i + 1:]
                dfs(new_nums, new_ops)

        # pattern要按ascii码
        nums = re.split('[*+-]', expression)
        ops = []
        for c in expression:
            if c in operators:
                ops.append(c)
        dfs(nums, ops)
        
        return [eval(exp) for exp in s]

总结

dfs来选可以选的地方
re.split来区分符号和数字，主要要根据ascii码的顺序
然后最后通过eval计算所有不同式子的结果即可