Unique Binary Search Trees--LeetCode

ゞ 浴缸里的玫瑰 2022-08-05 14:18 271阅读 0赞

题目:

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.

  1. 1 3 3 2 1
  2. \ / / / \ \
  3. 3 2 1 1 3 2
  4. / / \ \
  5. 2 1 2 3

思路:使用任意一个数值作为根节点,那么当前可以构成的数的总和就是左边数的个数乘以右边树的个数

这道题要求可行的二叉查找树的数量,其实二叉查找树可以任意取根,只要满足中序遍历有序的要求就可以。从处理子问题的角度来看,选取一个结点为根,就把结点切成左右子树,以这个结点为根的可行二叉树数量就是左右子树可行二叉树数量的乘积,所以总的数量是将以所有结点为根的可行结果累加起来。

熟悉 卡特兰数的朋友可能已经发现了,这正是 卡特兰数的一种定义方式,是一个典型的动态规划的定义方式(根据其实条件和递推式求解结果)。所以思路也很明确了,维护量res[i]表示含有i个结点的二叉查找树的数量。根据上述递推式依次求出1到n的的结果即可。
时间上每次求解i个结点的二叉查找树数量的需要一个i步的循环,总体要求n次,所以总时间复杂度是O(1+2+…+n)=O(n^2)。空间上需要一个数组来维护,并且需要前i个的所有信息,所以是O(n)。代码如下:

  1. public int numTrees(int n) {
  2. if(n<=0)
  3. return 0;
  4. int[] res = new int[n+1];
  5. res[0] = 1;
  6. res[1] = 1;
  7. for(int i=2;i<=n;i++)
  8. {
  9. for(int j=0;j<i;j++)
  10. {
  11. res[i] += res[j]*res[i-j-1];
  12. }
  13. }
  14. return res[n];
  15. }

也可以使用基本的递归思想,使用每一个数值作为根几点,左右可能树的总数的乘积

  1. int UniqueBST(int n)
  2. {
  3. if(n <= 1)
  4. return 1;
  5. else
  6. {
  7. int sum =0,i;
  8. for(i=1;i<=n;i++)
  9. {
  10. sum += UniqueBST(i-1)*UniqueBST(n-i);
  11. }
  12. return sum;
  13. }
  14. }

发表评论

表情:
评论列表 (有 0 条评论,271人围观)

还没有评论,来说两句吧...

相关阅读