题目描述:
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;
}