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);
    }
};