不同的二叉搜索树系列 【递归】

Leetcode 96. 不同的二叉搜索树

问题描述

给定一个整数 n，求以 1 … n 为节点组成的二叉搜索树有多少种？

解题报告

完全模仿 Leetcode 95 来实现的。

实现代码

class Solution {
public:
    int helper(int start,int end){
        int ret=0;
        if(start > end)
            return 1;
        
        for(int i=start;i<=end;i++){
            int left = helper(start,i-1);
            int right = helper(i+1,end);

            ret+=left*right;
        }
        return ret;
    }
    
    int numTrees(int n) {
        int ret=0;
        if(n == 0)
            return ret;
        return helper(1,n);
    }
};

Leetcode 95. 不同的二叉搜索树 II

问题描述

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

解题报告

求 1……n 的所有的二叉搜索树，有以下可能：

• 将 1 作为根节点，[] 作为左子树，[2, ……，n] 组成的所有可能的二叉搜索树作为右子树；
• 将 2 作为根节点，[1] 作为左子树，[3, ……，n] 组成的所有可能的二叉搜索树作为右子树；
• ……
• 将 n 作为根节点，[1,……n-1] 的所有可能二叉搜索树作为左子树，[] 作为右子树。
  根据上面的分析，可以使用递归实现。

实现时较为巧妙的一点是：

if(start>end) ret.push_back(nullptr);

这样，当调用到 helper(1,1) 时，left 为 [nullptr] 【存在一个元素的】，同理 right 为 [nullptr] 【存在一个元素】。
所以

for(auto l:left){
	for(auto r:right){
		TreeNode* root = new TreeNode(i);
        root -> left = l;
        root -> right = r;
        ret.push_back(root);
	}
}

经过这一步，会得到以 1 为根节点，左子树和左子树均为 null 的二叉搜索树。

实现代码

class Solution {
public:
    vector<TreeNode*> helper(int start,int end){
        vector<TreeNode*> ret;
        if(start > end)
            ret.push_back(nullptr);
        
        for(int i=start;i<=end;i++){
            vector<TreeNode*> left = helper(start,i-1);
            vector<TreeNode*> right = helper(i+1,end);

            for(auto l : left){
                for(auto r : right){
                    TreeNode* root = new TreeNode(i);
                    root -> left = l;
                    root -> right = r;
                    ret.push_back(root);
                }
            }
        }
        return ret;
    }
    
    vector<TreeNode*> generateTrees(int n) {
        vector<TreeNode*> ret;
        if(n == 0)
            return ret;    
        ret = helper(1,n);
        return ret;
    }
};

// 作者：black-hole
// 链接：https://leetcode-cn.com/problems/unique-binary-search-trees-ii/solution/c-30xing-dai-ma-di-gui-by-black-hole/
// 来源：力扣（LeetCode）
// 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

总结

递归法解决二叉树问题。

参考资料

[1]Leetcode 96. 不同的二叉搜索树
[2]Leetcode 95. 不同的二叉搜索树 II
[3]Leetcode 95. 题解区：black-hole
[4] 95*. Unique Binary Search Trees II