LeetCode题解(0823)：带因子的二叉树(Python)

最新推荐文章于 2024-09-21 10:02:17 发布

长行

最新推荐文章于 2024-09-21 10:02:17 发布

阅读量234

点赞数

分类专栏：刷题记录文章标签： leetcode python 动态规划树

本文链接：https://blog.csdn.net/Changxing_J/article/details/112767592

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

本文探讨了一道关于使用动态规划解决的题目，涉及树结构和因子计数。解法通过构造dp数组，计算不同因子组合下二叉树的结构数量，时间复杂度为O(N^2)，空间复杂度为O(N)。展示了Python代码实现及其执行效率。

摘要由CSDN通过智能技术生成

题目：原题链接（中等）

标签：动态规划、树

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O(N^2)$	$O (N)$	524ms (35.00%)
Ans 2 (Python)
Ans 3 (Python)

解法一：

class Solution:
    _MOD = 10 ** 9 + 7

    def numFactoredBinaryTrees(self, A: List[int]) -> int:
        size = len(A)
        A.sort()

        count = {n: i for i, n in enumerate(A)}

        dp = [1] * size
        for i in range(size):
            v = A[i] * A[i]
            if v in count:
                k = count[v]
                dp[k] += dp[i] * dp[i]
            for j in range(i):
                v = A[i] * A[j]
                if v in count:
                    k = count[v]
                    dp[k] += dp[i] * dp[j] * 2
        return sum(dp) % self._MOD