原题链接:
95. Unique Binary Search Trees II
【思路】
这是一道经典的分治算法题目,相比回溯算法,虽然都是递归中包含循环,循环中嵌套递归。但是,分治算法会利用到前面相互独立的子问题的中间结果,并将子问题层层串接在一起。最外层循环,将子问题划分为和为 n 的两个子问题,然后下一层递归又分别将子问题划分为更小的子问题,直到小问题小到只有大小为1的子问题,再将小问题的结果逐层向上回送。
public class Solution {
public List<TreeNode> generateTrees(int n) {
if(n <= 0) return new ArrayList<>();
return greateSubTree(1, n);
}
private List<TreeNode> greateSubTree(int start, int end) {
List<TreeNode> res = new ArrayList<>();
if(end < start) {
res.add(null);
return res;
}
for(int i = start; i <= end; i++) {
for(TreeNode l : greateSubTree(start, i-1)) { //
for(TreeNode r : greateSubTree(i+1, end)) {
TreeNode root = new TreeNode(i);
root.left = l;
root.right = r;
res.add(root);
}
}
}
return res;
}
}
9 / 9
test cases passed. Runtime: 6 ms Your runtime beats 3.38% of javasubmissions.