DFS/BFS专题1 - leetcode241. Different Ways to Add Parentheses/279. Perfect Squares - Mark

241. Different Ways to Add Parentheses

题目描述

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

例子

示例 1:
输入: “2-1-1”
输出: [0, 2]
解释:
((2-1)-1) = 0
(2-(1-1)) = 2

示例 2:
输入: “23-45”
输出: [-34, -14, -10, -10, 10]
解释:
(2*(3-(45))) = -34
((23)-(45)) = -14
((2(3-4))5) = -10
(2((3-4)5)) = -10
(((23)-4)*5) = 10

思想
分治法
1）递归地将表达式按照运算符拆分为左右两半；
2）分解成子问题，用操作符继续分成左右两部分；
3）截止条件：isdigit()
可采用dic存储优化

解法

class Solution(object):
    def diffWaysToCompute(self, input):
        """
        :type input: str
        :rtype: List[int]
        """
        self.dic = {}
        return self.helper(input)
    
    def helper(self, input):
        if input in self.dic:
            return self.dic[input]
        if input.isdigit():
            return [int(input)]
        
        res = []
        for i in range(len(input)):
            if input[i] in '+-*':
                l = self.diffWaysToCompute(input[:i])
                r = self.diffWaysToCompute(input[i+1:])
                
                for x in l:
                    for y in r:
                        if input[i] == '+':
                            res.append(x + y)
                        elif input[i] == '-':
                            res.append(x - y)
                        else:
                            res.append(x * y)
        self.dic[input] = res
        return res

279. Perfect Squares

题目描述

给定正整数 n，找到若干个完全平方数（比如 1, 4, 9, 16, …）使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

例子

示例 1:
输入: n = 12
输出: 3
解释: 12 = 4 + 4 + 4.

示例 2:
输入: n = 13
输出: 2
解释: 13 = 4 + 9.

思想
法1：DP

看到 “最小” 之类的词，考虑DP。

dp[i] 表示组成和为 i 的完全平方数的最小个数。
dp[i] = min(dp[i], dp[i-j*j])

法2：BFS
结点表示剩余的和，边表示完全平方数

解法1
复杂度 - 时间O(n^2)，空间O(n)

class Solution(object):
    def numSquares(self, n):
        """
        :type n: int
        :rtype: int
        """
        dp = [i for i in range(n+1)]
        for i in range(1, n+1):
            for j in range(1, int(i**0.5)+1):
                dp[i] = min(dp[i], dp[i-j*j]+1)
        return dp[-1]

解法2
BFS

class Solution(object):
    def numSquares(self, n):
        """
        :type n: int
        :rtype: int
        """
        depth = 0
        nodes = set([n])
        edges = [i * i for i in range(1, int(n**0.5)+1)]
        while edges:
            depth += 1
            nextLevel = set()
            for node in nodes:
                for edge in edges:
                    if edge == node:
                        return depth
                    elif edge < node:
                        nextLevel.add(node - edge)
                    else:
                        break
            nodes = nextLevel
        return depth

Ref

Summary of 4 different solutions (BFS, DP, static DP and mathematics)