算法记录
LeetCode 题目:
从左向右遍历一个数组,通过不断将其中的元素插入树中可以逐步地生成一棵二叉搜索树。
说明
一、题目
给定一个由不同节点组成的二叉搜索树,输出所有可能生成此树的数组。
二、分析
- 按照二叉搜索树的性质来说,只要确定了根节点 ,子节点随便插都会自动的平衡到对应的位置上,也就是和顺序无关。
- 那就需要在确定根的情况下对子节点进行全排列添加。
class Solution {
public List<List<Integer>> BSTSequences(TreeNode root) {
List<List<Integer>> ret = new ArrayList();
if(root == null) {
ret.add(new ArrayList());
return ret;
}
List<Integer> base = new ArrayList();
base.add(root.val);
List<List<Integer>> l_ = BSTSequences(root.left);
List<List<Integer>> r_ = BSTSequences(root.right);
for(List<Integer> l : l_) {
for(List<Integer> r : r_) {
List<Integer> buff = new ArrayList(base);
merge(buff, l, 0, r, 0, ret);
}
}
return ret;
}
public void merge(List<Integer> buff,
List<Integer> l, int l1,
List<Integer> r, int r1,
List<List<Integer>> ret) {
if(l1 == l.size() && r1 == r.size()) {
ret.add(new ArrayList(buff));
return ;
}
if(l1 < l.size()) {
buff.add(l.get(l1));
merge(buff, l, l1 + 1, r, r1, ret);
buff.remove(buff.size() - 1);
}
if(r1 < r.size()) {
buff.add(r.get(r1));
merge(buff, l, l1, r, r1 + 1, ret);
buff.remove(buff.size() - 1);
}
}
}
总结
熟悉二叉树的数据性质。