回溯算法-leetcode#96-给数字1-n建不同的二叉搜索树

最新推荐文章于 2022-05-12 11:24:14 发布

沙扬娜拉的裙裾

最新推荐文章于 2022-05-12 11:24:14 发布

阅读量124

点赞数

分类专栏：回溯算法 leetcode

本文链接：https://blog.csdn.net/u011089927/article/details/107607070

版权

leetcode 同时被 2 个专栏收录

88 篇文章 0 订阅

订阅专栏

回溯算法

11 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        vector<TreeNode*> res;
        if(n<=0) return res;
        res = build(1,n);
        return res;
    }
    vector<TreeNode*> build(int begin,int end){
        vector<TreeNode*> res;
        if(begin>end) {res.push_back(NULL);return res;}
        for(int i=begin;i<=end;i++){
            vector<TreeNode*> left = build(begin,i-1);
            vector<TreeNode*> right = build(i+1,end);
            for(TreeNode* l:left){
                for(TreeNode * r:right){
                    TreeNode * root=new TreeNode(i);
                    root->left = l;
                    root->right = r;
                    res.push_back(root);
                }
            }
        }
        return res;
    }
};

与95很类似，计算树种类时，建树种数只跟数字个数有关，在建树时，顺序遍历begin到end之间的所有位置，以每一个位置作为根，左侧建左子树，右侧建右子树，然后循环与根连接，加入结果集合即可。