LeetCode95 Unique Binary Search Trees II

题目链接：

https://leetcode.com/problems/unique-binary-search-trees-ii/

题目描述：

找出结点个数为n的所有二叉查找树。结点的val与1~n一一对应。

上次那道题是找出多少个就行了，T-T这次这个是要把找出来的树全部展示出来。

题目分析：

卡了好久，确实不会做，最后看了别人的代码。

http://www.2cto.com/kf/201305/215967.html

与前面那道题还是有相似之处的。从1~n依次选为根结点。

以i为根结点的树，其左子树由[1,i-1]构成，其右子树由[i+1,n]构成。

以i为根结点的树的种数等于其左子树的种数乘右子树的种数。

vector<TreeNode*> leftTrees代表左子树
vector<TreeNode*> rightTrees代表右子树

递归思想，依次选择根结点，对左右子序列再分别建树。

代码：

class Solution {
public:
vector<TreeNode*> getAllTrees(int beg,int end){
vector<TreeNode*> result;
if(beg>end){
result.push_back(NULL);
return result;
}
else{
for(int i=beg;i<=end;i++){//依次选择根结点
vector<TreeNode*> leftTrees=getAllTrees(beg,i-1);
vector<TreeNode*> rightTrees=getAllTrees(i+1,end);//对左右子序列再分别建树
for(int j=0;j<leftTrees.size();j++){
for(int k=0;k<rightTrees.size();k++){
//选择以i为根结点的树的个数等于其左右子树个数的乘积
TreeNode* root=new TreeNode(i);
root->left=leftTrees[j];
root->right=rightTrees[k];
result.push_back(root);
}
}
}
}
return result;
}
vector<TreeNode*> generateTrees(int n) {
vector<TreeNode*> result;
if(n<=0){
return result;
}
return getAllTrees(1,n);
}
};