Unique Binary Search Trees II

题目描述：

Given an integer n, generate all structurally unique BST's (binary search trees) that store values 1 ... n.

Example:
Input: 3
Output:
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
Explanation:
The above output corresponds to the 5 unique BST's shown below:

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \

   2     1         2                 3

思路:依次将每一个数字作为根节点，之后由该点划分左右子树同时对左右子树做相同的操作，最后即可得到结果。

#include<cstdio>
#include<vector>
using namespace std;

struct TreeNode 
{
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

vector<TreeNode*> Generate(int start,int end)
{
	vector<TreeNode*> result;
	//叶子节点直接放入vector中
	//虽然不写这个if判断下面也可以对叶子节点作出处理，但是直接处理可以提高整个算法运行速度 
	if(start==end)
	{
	    TreeNode *node = new TreeNode(start);
	    result.push_back(node);
	    return result;
	}
	for(int i=start;i<=end;i++)
	{ 
	    vector<TreeNode*> left;
	    vector<TreeNode*> right;
	    //对临界点，即start和end两个端点进行处理，start没有左子树，end没有右子树
	    //所以要对start的左子树和end的右子树直接赋值NULL
	    //若不处理，start的left和end的start均为0,直接导致下面两个for循环无法进行，无法构建树 
	    if(i==start)
	        left.push_back(NULL);
	    else
                left = Generate(start,i-1);
            if(i==end)
                right.push_back(NULL);
	    else
                right = Generate(i+1,end);
            //对构成左右子树进行组合 
	    for(int m=0;m<left.size();m++)
	    {
	    	for(int n=0;n<right.size();n++)
	        {
	    	    TreeNode *node = new TreeNode(i);
	    	    node->left = left[m];
	    	    node->right = right[n];
		    result.push_back(node); 				
	    	}
	    }
	}		     
	return result;
}
vector<TreeNode*> generateTrees(int n) 
{
	return Generate(1,n);
}
int main()
{
	int n = 3;
	vector<TreeNode*> result = generateTrees(n);
	for(int i=0;i<result.size();i++)
	    printf("%d ",result[i]->val);
	return 0;
}