码迷,mamicode.com
首页 > 编程语言 > 详细

95题--不同的二叉搜索树II(java、中等难度)

时间:2020-06-07 16:21:54      阅读:49      评论:0      收藏:0      [点我收藏+]

标签:示例   结构   有一个   之间   src   难度   href   不同   循环   

题目描述:给定一个整数 n,生成所有由 1 ... n 为节点所组成的 二叉搜索树 。

示例如下:

技术图片

 

  

  分析:这一题需要对比LeetCode96题来分析:https://www.cnblogs.com/KongJetLin/p/13054624.html

  第96题也是求所有由 1 ... n 为节点所组成的 二叉搜索树,但是96题返回的是所有二叉搜索树的数量,因此对于96题,我们只需要使用动态规划的方法,从n=2开始,根据:dp[n] = dp[0] * dp[n-1] + dp[1] * dp[n-2] + ...... + dp[i-1]*dp[n-i] + ...... + dp[n-2] * dp[1] + dp[n-1] * dp[0] 公式,逐个求dp[i] (i = 0,1,...,n),直到找到 dp[n] ,就是所有可能结构的二叉搜索树的数量。

  但是这一题95题,返回的是所有可能的二叉搜索树的根结点的集合 List<TreeNode>,我们可以使用递归的思想进行求解。

  求 1...n 的所有可能的二叉搜索树,可以分解为:

1)把 1 作为根节点,[ ] 空作为左子树,[ 2 ... n ] 的所有可能结构作为右子树;

2)2 作为根节点,[ 1 ] 作为左子树,[ 3...n ] 的所有可能结构作为右子树;

3)3 作为根节点,[ 1 2 ] 的所有可能作为左子树,[ 4 ... n ] 的所有可能作为右子树,然后左子树和右子树以3作为根结点进行两两组合。

...

3)n 作为根节点,[ 1... n ] 的所有可能作为左子树,[ ] 作为右子树。 

  对于求 [ 2 ... n ] 、 [ 3 ... n ] 、[ 1 2 ]等情况的所有可能,也可以利用上边的方法,把每个数字作为根节点,然后把所有可能的左子树和右子树组合起来即可。

  特殊情况,遇到 [ ] ,说明当前结点为null,以当前结点为根的所有可能子树只有一个,那就是 null,直接返回null;遇到 [ 1 ] 等只有一个数字的情况,当前结点为该数字,且当前结点的左右子树都没有结点(为null),那么以当前结点为根的所有可能子树只有一个,那就是当前数字,把该数字作为一棵树返回。

  详细的过程见下面代码分析:

public ArrayList<TreeNode> generateTrees (int n)
    {
        ArrayList<TreeNode> res = new ArrayList<>();

        /*
        当 n=0 的时候,不返回null,而是直接返回空的ArrayList<TreeNode>。
        因为题目要求返回类型是ArrayList<TreeNode>,返回null会出错
         */
        if(n == 0)
            return res;
        //查找 1 - n的数字所组成的所有可能结构的二叉树
        return generateTrees(1, n);
    }

    //查找 start到end 数字所组成的所有可能结构的二叉树,并将这些二叉树的根放入ArrayList<TreeNode>
    private ArrayList<TreeNode> generateTrees(int start , int end)
    {
        //这个集合用于存储所有可能结构的二叉树的根(因为可能的二叉树结构有多个,因此要用ArrayList保存这些二叉树的根结点)
        ArrayList<TreeNode> res = new ArrayList<>();

        /*
        可能遇到 start>end 的情况,如上一层递归以start为根,其左子树为generateTrees(start, start - 1),
        即左子树没有结点,为null,那么所有可能的二叉树只有一个:null;
        可能遇到 start=end 的情况,即已经找到最大树的叶子结点,叶子结点没有左右子树,
        那么所有可能的二叉树只有一个:start/end 为根。

        上述情况分别对应1结点的左子树(null)以及右子树(2)的情况。
                    1
                  /                 null   2
         */
        if(start > end)
        {
            res.add(null);
            return res;//注意返回集合,因为上一层的递归遍历的是所有可能的二叉子树的根结点的集合
        }
        if(start == end)
        {
            res.add(new TreeNode(start));
            return res;
        }


        //第一个for循环用于遍历 start到end 之间所有的数字,求以这些数字为根的二叉树
        for (int i = start; i <= end ; i++)
        {
            //求 start到i-1 数字组成的所有可能的二叉树,将这些二叉树根的集合返回,这些二叉树是以i为根的树的左子树
            ArrayList<TreeNode> leftTrees = generateTrees(start, i - 1);
            ArrayList<TreeNode> rightTrees = generateTrees(i + 1, end);//类似于求左子树的操作

            /**
            leftTrees 与 rightTrees分别存储所有可能的左子树与右子树的根结点。
             我们需要将这些左子树与右子树两两组合,并将i作为当前树的根,连接这些左右子树,最后将这个根i添加到 res。

             以i为根的树可能有多种结构,而我们通过最外层的循环,找到以 i=start-end 为根的所有可能树的结构,
             将这些结构的根添加到 res 集合中,返回给上一层的递归即可。
             */
            for (TreeNode leftTree : leftTrees)
            {
                for (TreeNode rightTree : rightTrees)
                {
                    TreeNode root = new TreeNode(i);//以i作为根结点
                    //根结点连接所有可能的左右子树,i根结点若连接不同的左右子树,最后测试的时候代表其是不同的二叉树
                    root.left = leftTree;
                    root.right = rightTree;
                    //将这个结构的二叉搜索树的根添加到 res
                    res.add(root);
                }
            }
        }
        //出循环的时候,将以 i=start-end 为根的所有可能树的结构的根结点添加到 res,将res集合返回给上一层递归
        return res;

        /**
        注意,这里的返回值是ArrayList<TreeNode>,而不是 ArrayList<ArrayList<TreeNode>>,
         因此我们需要查找每一个i(i=1,2,...,n)为根的所有可能的二叉树结构,并将这些根(i)结点添加到ArrayList<TreeNode>。

         如果返回 ArrayList<ArrayList<TreeNode>>,就要找到所有可能的二叉树的所有结点的集合,将这些ArrayList<TreeNode>
         再次封装到 ArrayList 中。
         */
    }

 

95题--不同的二叉搜索树II(java、中等难度)

标签:示例   结构   有一个   之间   src   难度   href   不同   循环   

原文地址:https://www.cnblogs.com/KongJetLin/p/13060932.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!