LeetCode95. 不同的二叉搜索树 II

Posted 2020-11-12 masteryan576356467

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

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     vector<TreeNode*> generateTrees(int n) {
13         if(n) return dfs(1, n);
14         return {};
15     }
16     
17     vector<TreeNode *> dfs(int l, int r){
18         vector<TreeNode *> ans;
19         if(l > r){//递归出口
20             ans.push_back(nullptr);
21             return ans;
22         }
23         for(int i = l; i <= r; ++i){    　　　　　　　　//以当前数字i为根
24             vector<TreeNode *> left = dfs(l, i - 1);   //获取左孩子
25             vector<TreeNode *> right = dfs(i + 1, r);　　//获取有孩子
26             for(auto &ii : left){　　　　　　　　　　　　　　//提取左孩子
27                 for(auto &jj : right){　　　　　　　　　　　　//提取右孩子
28                     TreeNode *t = new TreeNode{i};
29                     t->left = ii;
30                     t->right = jj;
31                     ans.push_back(t);
32                 }
33             }
34         }
35         return ans;
36     }
37     
38 };

分析：

答案的数量应该是卡特兰数，

公式 

递归方法建树