LeetCode笔记:96. Unique Binary Search Trees

2021-11-23 14:10:28 浏览数 (1)

问题:

Given n, how many structurally unique BST's (binary search trees) that store values 1...n? For example, Given n = 3, there are a total of 5 unique BST's.

image.png

大意:

给出n,包含1~n这些节点可以形成多少个不同的BST(二叉查找树)? 比如, 给出n = 3,有5个不同的BST:

image.png

思路:

二叉查找树的性质是左子节点一定小于父节点,右子节点一定大于父节点。

我们思考一下可以发现,要形成不同的二叉树,最基本的分类是1n各自都做一次根节点。在它们作为根节点时,又分别还有多少种不同的组合方式呢?由于这是一个二叉查找树,那么根节点的左边一定都是小于他的数,右边一定都是大于它的数,所以1n就会被分成两部分去放置,这时候由可以分别把左子节点、右子节点分别看成要安放一部分数字的根节点,又变成了一样的规律。

所以假设以i为根节点,可能的组合情况为F(i,n),而G(n)为输入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),我们必须把前面的数都计算出来。

代码(Java):

代码语言:javascript复制
public class Solution {
    public int numTrees(int n) {
        int[] res = new int[n 1];
        res[0] = 1;
        res[1] = 1;
        for (int i = 2; i <= n; i  ) {
            for (int j = 1; j <= i; j  ) {
                res[i]  = res[j-1] * res[i-j];
            }
        }
        return res[n];
    }
}

合集:https://github.com/Cloudox/LeetCode-Record

查看作者首页

0 人点赞