Day23-Task95.不同的二叉搜索树Ⅱ

题目描述

95.不同的二叉搜索树Ⅱ

给定一个整数 n，生成所有由 1 … n 为节点所组成的二叉搜索树。

示例

输入: 3
输出:
[
[1,null,3,2],
[3,2,null,1],
[3,1,null,null,2],
[2,1,3],
[1,null,2,null,3]
]
解释:
以上的输出对应以下 5 种不同结构的二叉搜索树：

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \
   2     1         2                 3

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/unique-binary-search-trees-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

二叉查找树（Binary Search Tree），（又：二叉搜索树，二叉排序树）它或者是一棵空树，或者是具有下列性质的二叉树： 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值； 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值； 它的左、右子树也分别为二叉排序树。

基于二叉搜索树的概念，看到这道题的第一思路是，从1到n，分别令每个值当树根建立二叉搜索树。但是在实现过程中发现并不是这么简单。

由于对于每一个 val=i 的根节点，其建立的二叉搜索树并不是唯一的，还与数据输入的顺序有关。因此采用上述方法不成立。

参考题解后，发现对于整数n，由 1 … n 为节点所组成的二叉搜索树是有规律的。这种规律符合卡特兰数的规律。

卡特兰数又称卡塔兰数，英文名Catalan number，是组合数学中一个常出现在各种计数问题中出现的数列。以比利时的数学家欧仁·查理·卡塔兰 (1814–1894)的名字来命名，其前几项为（从第零项开始） : 1, 1, 2, 5, 14, 42, 132, 429, 1430, 4862, 16796, 58786, 208012, 742900, 2674440, 9694845, 35357670, 129644790, 477638700, 1767263190, 6564120420, 24466267020, 91482563640, 343059613650, 1289904147324, 4861946401452, …

卡特兰数Cn满足以下递推关系：

而由 1 … n 为节点所组成的二叉搜索树G(n)则满足一定递推规律。参考96.不同的二叉搜索树题解。

对于以 i 为根节点的二叉树，可产生G(i-1)种左子树，G(n-i)种右子树。因此可以根据这个规律，分别递归建立二叉搜索树的左右子树，并在最后将其合并到根节点 root 中

• 若 n==0 直接返回空集
• 若 n!=0 开始递归建树

时间复杂度： O((4ⁿ)/(n^(1/2)))

代码如下

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<TreeNode*> generate(int start, int end)
    {
      //建立全部二叉搜索树集合all_trees
        vector<TreeNode*> all_trees;
        //递归返回条件，start>end时，递归终止
        if(start>end)
        {
            all_trees.push_back(NULL);
            return all_trees;
        }
        else
        {
          //建立左子树结点为i-1，右子树结点为n-i的全部子树
            for(int i=start;i<=end;i++)
            {
                vector<TreeNode*> left = generate(start,i-1);
                vector<TreeNode*> right = generate(i+1,end);
                //遍历左子树集合
                for(auto l : left)
                {
                  //遍历右子树集合
                    for(auto r : right)
                    {
                      //将建立完成的左右子树加到以val==i的根结点上
                        TreeNode* current_tree = new TreeNode(i);
                        current_tree->left = l;
                        current_tree->right = r;
                        //将生成的二叉搜索树加到all_trees中
                        all_trees.push_back(current_tree);
                    }
                }
            }
        }
      //返回全部二叉搜索树集合all_trees
        return all_trees;
    }
    vector<TreeNode*> generateTrees(int n) {
      //若n==0，直接返回空集
        if(n==0)
        {
            return {};
        }
        else
        {
            return generate(1,n);
        }
    }
};

执行结果