241. Different Ways to Add Parentheses——本质:DFS

Given a string of numbers and operators, return all possible results from computing all the different possible ways to group numbers and operators. The valid operators are +, - and *.


Example 1

Input: "2-1-1".

((2-1)-1) = 0
(2-(1-1)) = 2

Output: [0, 2]


Example 2

Input: "2*3-4*5"

(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

Output: [-34, -14, -10, -10, 10]

class Solution(object):
    def diffWaysToCompute(self, input):
        """
        :type input: str
        :rtype: List[int]
        A: (2*3)-4*5
        ((2*3)-4)*5=>1
        (2*3)-(4*5)=>2
        B:
        2*(3-4)*5
        (2*(3-4))*5=>1
        2*((3-4)*5)=>2
        """
        import re
        tokens = re.split('(\D)', input)
        nums = map(int, tokens[::2])
        ops = map({'+': operator.add, '-': operator.sub, '*': operator.mul}.get, tokens[1::2])
        def build(lo, hi):
            if lo == hi:
                return [nums[lo]]
            ans = []
            for i in range(lo, hi):
                for a in build(lo, i):
                    for b in build(i+1, hi):
                        ans.append(ops[i](a, b))    
            return ans
        return build(0, len(ops))        

 

re.split("(\D)", "2+3-1")
['2', '+', '3', '-', '1']

posted @ 2016-12-29 23:02  bonelee  阅读(296)  评论(0编辑  收藏  举报