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

题目链接

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的事情

posted @ 2020-07-21 10:22  ZJPang  阅读(92)  评论(0编辑  收藏  举报