LeetCode-二叉搜索树序列

148 阅读1分钟

算法记录

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);
        }             
    }
}

总结

熟悉二叉树的数据性质。