Day 41 | 动态规划 03

67 阅读1分钟

343. 整数拆分

注意麻烦的是n=2n=3时,dp[2]=1,但更新后续状态时 需要取原值

def integerBreak(self, n: int) -> int:
    dp = [i-1 for i in range(n+1)]
    dp[0] = 1
    dp[1] = 1
    for i in range(2, n+1):
        for j in range(2, i-1):
            dp[i] = max(dp[i], max(j,dp[j])*max(i-j,dp[i-j]))
    
    return dp[-1]

96. 不同的二叉搜索树

参考了随想录的思路

元素1为头结点搜索树的数量 = 右子树有2个元素的搜索树数量 * 左子树有0个元素的搜索树数量

元素2为头结点搜索树的数量 = 右子树有1个元素的搜索树数量 * 左子树有1个元素的搜索树数量

元素3为头结点搜索树的数量 = 右子树有0个元素的搜索树数量 * 左子树有2个元素的搜索树数量

dp[3] = dp[2] * dp[0] + dp[1] * dp[1] + dp[0] * dp[2]

def numTrees(self, n: int) -> int:
    dp = [0] * (n+1)
    dp[0] = dp[1] = 1
    for i in range(2,n+1):
        for j in range(1,i+1):
            # 讨论上一个节点形成的
            dp[i] += dp[j-1] * dp[i-j]
    return dp[-1]