题目描述:
给定一个整数 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
思路分析:
树的题目,首先就是用递归思想来做。自己一开始的做法,是不停的去拷贝对象,这样容易出错,且很繁琐。参考了别人的代码,直接用指针简洁了很多。
代码:
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*> generateTreesCore(int l, int r)
13 {
14 vector<TreeNode*>ans;
15 if(l>r)
16 ans.push_back(NULL);
17 else
18 {
19 for(int i=l; i<=r; i++)
20 {
21 vector<TreeNode*> left = generateTreesCore(l, i-1);
22 vector<TreeNode*> right = generateTreesCore(i+1, r);
23 for(int idx_l=0; idx_l<left.size(); idx_l++)
24 {
25 for(int idx_r=0; idx_r<right.size(); idx_r++)
26 {
27 TreeNode* cur = new TreeNode(i);
28 cur->left = left[idx_l];
29 cur->right = right[idx_r];
30 ans.push_back(cur);
31 }
32 }
33 }
34 }
35 return ans;
36 }
37 vector<TreeNode*> generateTrees(int n) {
38 vector<TreeNode*> res;
39 if(n<=0)
40 return res;
41 res = generateTreesCore(1, n);
42 return res;
43 }
44 };