96. Unique Binary Search Trees
1. 题目
题目链接
Given n, how many structurally unique BST’s (binary search trees) that store values 1 … n?
Example:
Input: 3
Output: 5
Explanation:
Given n = 3, there are a total of 5 unique BST’s:
2. 题目分析
题目要求是,给定n,用关键字为1~n的这些节点,可以形成多少中二叉查找树(BST)。
3. 解题思路
二叉查找树的性质是左子树的所有节点一定小于根节点,右子树的所有节点一定大于根节点。
根据BST的特点,一颗二叉查找树可以分为三部分:根节点、左子树、右子树。关键字1 ~ n都可以作为根节点。在它们作为根节点时,又分别还有多少种不同的组合方式呢?由于这是一个二叉查找树,那么根节点 i 的左子树一定都是小于他的数[1,i-1],右子树一定都是大于它的数[i+1,n]。然后分别对左子树和右子树进行相同的分割,直到只有叶子节点为止。
按照我之前分析的动态规划的思路:
- 首先确定状态,用dp[i]表示用关键字1…i的节点能组成的二叉查找树的种类;
- 初始化,对第0行的所以状态进行初始化,即dp[0] = 1;dp[1] = 1;
- 找状态转移方程,dp(n) = dp(0) * dp(n-1) + dp(1) * dp(n-2) + …… + dp(n-1) *dp(0)
状态转移方程分析如下:参考博客https://blog.csdn.net/Cloudox_/article/details/70236770
假设以i为根节点的种类情况为F(i,n),而G(n)为1…n的节点能组成的二叉查找树的种类。则
F(i,n) = G(i-1)* G(n-i)
也就是左子节点以下的可能数量乘以右子节点以下的可能数量。
而因为1~n都可能作为根节点,所以最终的值是它们的和,也就是
G(n) = F(1,n) + F(2,n) + …… +F(n,n)
换算一下就是
G(n) = G(0) * G(n-1) + G(1) * G(n-2) + …… + G(n-1) *G(0)
其中我们可以直接看出 G(0) = G(1) = 1。这个作为初始值来递归计算就可以了,要知道G(n),我们必须把前面的数都计算出来。
4. 代码实现(java)
package com.algorithm.leetcode.dynamicAllocation;
/**
* Created by 凌 on 2019/1/26.
* 注释:96. Unique Binary Search Trees
*/
public class NumTrees {
public int numTrees(int n) {
int[] dp = new int[n+1];
dp[0] = 1;
dp[1] = 1;
for (int i = 2; i <= n; i++) {
for (int j = 1; j <= i; j++) {
dp[i] += (dp[j-1] * dp[i-j]);
}
}
return dp[n];
}
}