LeetCode #95 - Unique Binary Search Trees II

最新推荐文章于 2021-09-15 15:18:50 发布

原创最新推荐文章于 2021-09-15 15:18:50 发布 · 123 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

625 篇文章

订阅专栏

本文介绍了一种生成所有可能的不同二叉搜索树(BST)的算法实现，通过递归方式构建从1到n的BST，并展示了具体的C++代码实现。

题目描述：

Given an integer n, generate all structurally unique BST's (binary search trees) that store values 1 ... n.

Example:

Input: 3
Output:
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
Explanation:
The above output corresponds to the 5 unique BST's shown below:

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

与Unique Binary Search Trees类似，只是要把所有不同的BST都找出来，所以在递归时要把节点值的范围也作为参数传递。

class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        return generateTrees_recur(1,n);
    }
    
    vector<TreeNode*> generateTrees_recur(int begin, int end)
    {
        vector<TreeNode*> result;
        if(begin==end)
        {
            TreeNode* root=new TreeNode(begin);
            result.push_back(root);
            return result;
        }
        for(int i=begin;i<=end;i++)
        {
            vector<TreeNode*> left;
            vector<TreeNode*> right;
            if(i==begin) left.push_back(NULL);
            else left=generateTrees_recur(begin,i-1);
            if(i==end) right.push_back(NULL);
            else right=generateTrees_recur(i+1,end);
            for(int j=0;j<left.size();j++)
            {
                for(int k=0;k<right.size();k++)
                {
                    TreeNode* root=new TreeNode(i);
                    root->left=left[j];
                    root->right=right[k];
                    result.push_back(root);
                }
            }
        }
        return result;
    }
};