LeetCode: Unique Binary Search Trees II

最新推荐文章于 2021-09-11 12:08:06 发布

原创最新推荐文章于 2021-09-11 12:08:06 发布 · 681 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种使用递归方法构建二叉搜索树(BST)的方法，重点在于如何根据指定范围内的节点数量确定左右子树的元素个数。通过递归地解决左右子树的问题，并组合这些子树来构造所有可能的BST。

思路：经典递归方法，注意左右子树元素的个数。

code:

class Solution {
public:
    vector<TreeNode *> solve(vector<int> &num,int start,int n){
        vector<TreeNode *> ret;
        if(n == 0){
            ret.push_back(NULL);
            return ret;
        }
        for(int i = start;i < start + n;i++){
            int nLeft = i - start;
            int nRight = n - nLeft - 1;
            
            vector<TreeNode *> leftRet, rightRet;
            leftRet = solve(num,start,nLeft);
            rightRet = solve(num,i+1,nRight);
                for(int j = 0;j < leftRet.size();j++){
                    
                    for(int k = 0;k < rightRet.size();k++){
                        TreeNode * root = new TreeNode(num[i]);
                        root->left = leftRet[j];
                        root->right = rightRet[k];
                        ret.push_back(root);
                    }
                }
        }
        return ret;
    }
    vector<TreeNode *> generateTrees(int n) {
        vector<int> num(n,0);
        for(int i = 0;i <n;i++)
            num[i] = i+1;
        vector<TreeNode *> ret;
        ret = solve(num,0,n);
        return ret;
    }
};