leetcode_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

思路

这道题的思路很简单，只有两个字——“枚举”！

枚举思路用递归实现比较好，所以我们必须要定义一个递归函数，此函数有两个参数——左端点 l、右端点 r ，表示着分配给这一层递归从 i 到 j 的数值建树。

边界条件：l>r则递归结束，返回空列表。

接下来是三重循环 i, j, k，我将一个个讲解：

• 第一层i：表示的是根节点的循环，以下再定义两个 TreeNode* 类型的数组，分别存储递归建立左子树和右子树的所有结果。
• 第二层j：可以理解为与第三层是并列的，j循环指向所有左子树的建立结果。
• 第三层k：循环指向所有右子树的建立结果。然后将其与第二层的指向元素一起，拼接到 root 上，存入答案中。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        if(n == 0) return {};
        else return helper(1, n);
    }
    vector<TreeNode*> helper(int l, int r) {
        if(l > r) return {nullptr};
        vector<TreeNode*> ans;
        for(int i = l; i <= r; ++ i) {
            vector<TreeNode*> ll = helper(l, i-1);
            vector<TreeNode*> rr = helper(i+1, r);
            for(int j = 0; j < ll.size(); ++ j) {
                for(int k = 0; k < rr.size(); ++ k) {
                    TreeNode* root = new TreeNode(i);
                    root -> left = ll[j];
                    root -> right = rr[k];
                    ans.push_back(root);
                }
            }
        }
        return ans;
    }
};