1. 问题描述:
对于一棵有根二叉树 T,小蓝定义这棵树中结点的权值 W(T) 如下:空子树的权值为 0。如果一个结点 v 有左子树 L, 右子树 R,分别有 C(L) 和 C(R) 个结点,则:
W(v) = 1 + 2W(L) + 3W(R) + (C(L)) ^ 2 C(R)。
树的权值定义为树的根结点的权值。小蓝想知道,对于一棵有 2021 个结点的二叉树,树的权值最小可能是多少?
2. 思路分析:
分析题目可以知道2021个节点的二叉树的形态是具有很多种的,我们需要在节点数为2021的所有二叉树中找出一棵二叉树使得W(v) = 1 + 2W(L) + 3W(R) + (C(L)) ^ 2 C(R)的值是最小的,分析这个式子可以知道这个式子其实是一个递推方程,从理论上我们可以使用递归尝试所有的可能的方案(尝试左右两边可能的节点情况)但是由于节点数太大了实际上是不可行的,本质上题目给出的这个式子就是状态转移方程,直接根据这个式子进行递推即可,我们可以从小范围的节点数目一直递推到大范围的节点数目,这样其实就是尝试所有可能的二叉树的所有形态了,最后返回数组中的最后一个节点的值即可。(递推或者是动态规划都可以看成是暴力枚举的优化)一般来说可以使用递归解决的问题我们不妨考虑一下是否可以使用递推或者是动态规划的思路解决。
3. 代码如下:
答案是2653631372
import sys
if __name__ == '__main__':
# 空节点的节点值为0
dp = [0]
for i in range(1, 2022):
dp.append(sys.maxsize)
for j in range(i):
# 左边节点j个右边节点i - 1 - j个
dp[i] = min(dp[i], 1 + 2 * dp[j] + 3 * dp[i - 1 - j] + j * j * (i - 1 - j))
print(dp[2021])