首页 > 其他 > 详细

LeetCode 95. 不同的二叉搜索树 II

时间:2020-07-21 12:37:41      阅读:67      评论:0      收藏:0      [点我收藏+]

题目链接

95. 不同的二叉搜索树 II

思路分析

这个题要求我们把树建模出来,那么我们只好对1-n这些数字一个一个进行二叉搜索树建模。
我们利用二叉搜索树的特性,左子树小于根节点、右子树大于根节点的特性进行建立。我们最外围使用1-n中的每一个数字作为根节点,然后递归生成其左子树和右子树。
但是在做这个题的时候我自己脑子有点傻逼了,因为对于每一个递归返回的list,我们都需要先选取左右子树中的其中一个,然后再把另外一边拼接上,如果左子树或者右子树其中一个为空,那么就会导致当前结点的树会被全部抛弃掉。
这个问题卡了我很TM久,然后我实在忍不住了就去看了评论区,发现他们在返回空list之前,往list中加入了null,直接避免了出现返回空list的情况(这个思路是真TMD牛逼
这里还需要注意下边界问题,n等于0的时候返回的是空list,而不是有一个空结点的list。

代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<TreeNode> generateTrees(int n) {
        if(n == 0){
            return new ArrayList<TreeNode>();
        }
        return helper(1,n);
    }

    private List<TreeNode> helper(int left, int right){
        List<TreeNode> res = new LinkedList<>();
        if(left > right){
            res.add(null);
            return res;
        }
        for(int i = left; i <= right; i++){
            List<TreeNode> leftNodes = helper(left, i-1);
            List<TreeNode> rightNodes = helper(i+1, right);
            for(TreeNode leftNode : leftNodes){
                for(TreeNode rightNode : rightNodes){
                    TreeNode root = new TreeNode(i);
                    root.left =  leftNode;
                    root.right = rightNode;
                    res.add(root);
                }
            }
        }
        return res;
    }
}

总结

适当利用空指针也是一件非常NB的事情

LeetCode 95. 不同的二叉搜索树 II

原文:https://www.cnblogs.com/ZJPaang/p/13353577.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!