95. 不同的二叉搜索树 II (C++)---递归解题

题目详情

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

 

示例：

输入：3
输出：
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
解释：
以上的输出对应以下 5 种不同结构的二叉搜索树：

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

提示：

• 0 <= n <= 8

 

——题目难度:中等


 

---

定义

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */

-代码如下

class Solution {
public:
	vector<TreeNode*> helper(int start, int end) {		
		if (start > end) return {NULL};
		vector<TreeNode*> ret;
		
		/*在区间 [start, end] 不断生成以 i(i为整数) 为根节点的二叉搜索树*/  
		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*> ans;
        if (n == 0) return ans;
        ans = helper(1, n);
        return ans;
    }
};

结果