Unique Binary Search Trees II

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

weixin_30625691

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

阅读量59

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/illfuckingkyzb/p/10217647.html

本文详细解析了一种用于生成二叉搜索树（BST）的递归算法，通过递归调用实现左右子树的创建，并在每个节点处进行根节点的替换，最终生成所有可能的BST组合。

class Solution {
public:
    
    vector<TreeNode*> generateTrees(int n) {
        if (n == 0) {
            vector<TreeNode*> ret;
            return ret;
        }
        return generateTrees(n, 1);//由于数列从1开始，所以传递1为参数。
    }
     vector<TreeNode*> generateTrees(int n, int val_start) {
        vector<TreeNode*> ret;
        if (n == 0) { 
            ret.push_back(NULL);
            return ret;
        }
        if (n == 1) {
            ret.push_back(new TreeNode(val_start));
            return ret;
        }
        
        for (int i = 0; i < n; i++) {
            vector<TreeNode*> left = generateTrees(i, val_start);//从左子树为0开始往下递归。
            vector<TreeNode*> right = generateTrees(n-1-i, val_start+i+1);//右子树的数量就为（n-1-i）  其中1为根。
            for (int l = 0; l < left.size(); l++) {
                for (int r = 0; r < right.size(); r++) {
                    TreeNode* root = new TreeNode(val_start+i);//在这里实现根节点的替换
                    root->left = left[l];
                    root->right = right[r];
                    ret.push_back(root);
                }
            }
        }
        
        return ret;
    }
};