题目链接:
https://leetcode.com/problems/unique-binary-search-trees-ii/
题目描述:
找出结点个数为n的所有二叉查找树。结点的val与1~n一一对应。
上次那道题是找出多少个就行了,T-T这次这个是要把找出来的树全部展示出来。
题目分析:
卡了好久,确实不会做,最后看了别人的代码。
与前面那道题还是有相似之处的。从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);
}
};