今日签到题,题目如下:
给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一开始我以为哪个数字在什么位置根本无所谓,对结果毫无影响,所以想了半天没有思路。
官方题解给出了两个方案:1.动态规划,2.数学
数学是直接使用卡塔兰数,还没仔细看,以下就不讨论了。
虽然我有想过用动态规划,但实在是没有找出 dp[i] 要从哪里转移,所以想了许久未果。
数字在什么位置的确是有用的,可以帮助我们规划状态。
如果总共有 n 个节点,依次遍历 1 到 n 的每个数字,并将它放在根节点,然后令它左边的数字都分布在左节点之下,右节点都分布在右节点之下。假设当前遍历到的数字是 i ,那么 i 左边数字二叉树种类数量,其实就是有 i - 1 个节点的二叉树种类数量。右边同理,即是 n - i 个节点的二叉树种类数量。
综上,n 个节点的状态转移自所有小于 n 个节点的二叉树种类数量,所以正序遍历 0 到 n,最后可以获得转移结果 n 的二叉树种类数量。
讨论边界条件,0 个节点不转移自任何状态,但是可知 0 个节点二叉树种类数量为 1。
复杂度分析:
需要枚举小于 n 的所有状态(包括 0),而每个状态都需要枚举从 1 到该数字作为根节点的状态,即嵌套两次长度为 n 的循环,时间复杂度为 O(N^2)。
空间复杂度,需要一个数组保存所有小于等于 n 的状态(包括0),故空间复杂度为 O(N)。
以下为自己提交的代码:
public class Solution {
public int NumTrees(int n) {
int[] dp = new int[n + 1];
dp[0] = 1;
for(int i = 1;i < n+1;i++)
{
for(int j = 1;j <= i ;j++)
{
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
}