15.不同的二叉搜索树
- 题目描述
给你一个整数 n
,求恰由 n
个节点组成且节点值从 1
到 n
互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3
输出:5
示例 2:
输入:n = 1
输出:1
- 思路分析
1.创建一维数组dp[i]:表示节点数为i时,有几种搜索二叉树
2.根据题目描述进行规律探寻:
dp[0] = 1
dp[1] = dp[0] * dp[0]
dp[2] = dp[1] * dp[0] + dp[0] * dp[1]
dp[3] = dp[2] * dp[0] + dp[1] * dp[1] + dp[0] * dp[2]
dp[4] = dp[0] * dp[3] + dp[1] * dp[2] + dp[2] * dp[1] + dp[3] * dp[0]
解释:(1作为节点:左子树0个节点,右子树3个节点)
解释:(2作为节点:左子树1个节点,右子树2个节点)
解释:(3作为节点:左子树2个节点,右子树1个节点)
解释:(4作为节点:左子树3个节点,右子树0个节点)
规律总结:不难看出,每次求dp[i]时,不同的二叉树形态总合为每个不同节点作为根节点时的左右子树呈现的形态数
设i为根节点,j为左子树节点的个数,由上述规律可以计算得到:dp[i] = dp[i] + dp[j] * dp[i - 1 - j]这一公式;
- Java代码实现
public static int numTrees(int n) {
//1.创建一维数组dp[i]:表示节点数为i时的卡特兰数
int[] dp = new int[n + 1];
//2.初始化数组
dp[0] = 1;
//3.递推公式
for (int i = 1; i <= n; i++) {
for (int j = 0; j < i; j++) {
dp[i] = dp[i] + dp[j] * dp[i - 1 - j];
}
}
return dp[n];
}