题目
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
问题分析
本题要求求出从1~n的所有的二叉搜索树,以每个数作为根节点,使用分治的方式左右划分子树,递归构造 刚开始将[1,n]看作一个整体,需要将其每个数作为根节点,划分整个节点的左右区间,调用递归函数获取这个数的所有的左右子树,使用二重循环将所有的左右子树拼接到这个数构成的节点上,然后向上层返回,代码如下
代码示例
public List<TreeNode> generateTrees(int n) {
if(n < 1){
return new ArrayList<TreeNode>();
}
return help(1, n);
}
/**
* 分治递归求解所有树
* @param start
* @param end
* @return
*/
private List<TreeNode> help(int start, int end) {
List<TreeNode> res = new ArrayList<TreeNode>();
// return null node
if(start > end){
res.add(null);
return res;
}
// 从start到end每个节点都作为root向上返回,分治思想
for(int i = start; i <= end; i++){
// 左右子树
List<TreeNode> left = help(start, i - 1);
List<TreeNode> right = help(i + 1, end);
// 将以该节点为root的所有左右子树拼接到该节点上,使用二重循环
for(TreeNode a : left){
for(TreeNode b : right){
TreeNode node = new TreeNode(i);
node.left = a;
node.right = b;
res.add(node);
}
}
}
return res;
}