leetcode 95. 不同的二叉搜索树II
题目描述:
给定一个整数 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 };