第一次在博客中写题目解题思路,内心比较激动。这题是我昨晚失眠在pad中写出来的,这里记录一下。
题目
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
- 示例:
输入: 3 输出: 5 解释: 给定 n = 3, 一共有 5 种不同结构的二叉搜索树: [1,null,3,2],[3,2,null,1],[3,1,null,null,2],[2,1,3],[1,null,2,null,3]
解题思路
二叉搜索树的概念:
二叉查找树(Binary Search Tree),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。
使用动态规划的思想。
- 考虑将每增加一个节点的二叉搜索树种类数量记录在一个一维数组里,给定整数n,则将其值记录到一个n+1维的数组nums里;
- 设置nums数组的初始值,nums[1] = 1;
- 数组中,从2,3,…, i, …, n位的转移:
1)当左孩子或右孩子节点为空时,有nums[i - 1]种组合种类;
2)左孩子有j个节点,右孩子有i - j - 1个节点,此时该节点组合的二叉搜索树共有nums[j] * nums[i - j - 1]个;
3)将所有1)2)中的组合种类数相加则得到整数i的二叉搜索树的组合种类统计个数。
实现代码
class Solution {
public int numTrees(int n) {
int[] count = new int[n + 1];//n+1维数组
count[0] = 0;
count[1] = 1;
for(int i = 2; i <= n; i++){
for(int j = 0; j < i; j++){
if(j == 0 || j == i - 1)
count[i] += count[i - 1]; //当左孩子或右孩子节点为空时,有nums[i - 1]种组合种类
else{
count[i] += count[j] * count[i - j - 1];//左右子树种类的笛卡尔积
}
}
}
return count[n];
}
}