代码随想录算法训练营第十六天 | 104.二叉树的最大深度、559.n叉树的最大深度、 111.二叉树的最小深度、222.完全二叉树的节点个数

学习目标

• 完成二叉树相关题目(递归法)

---

学习内容

104.二叉树的最大深度(Easy)

题目链接:104.二叉树的最大深度

题目：给定一个二叉树，找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例：
给定二叉树 [3,9,20,null,null,15,7]，
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。

思路：这题目在第十四天的练习中有迭代法, 这里使用递归法去计算.
时间复杂度：O(n) 空间复杂度：O(log n)(如果不认可栈空间为额外空间应该是 O(1), 但 log n 是栈的深度)

解决方案：

class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null) return 0;
        int leftDepth = 1 + maxDepth(root.left);
        int rightDepth = 1 + maxDepth(root.right);
        return leftDepth > rightDepth ? leftDepth : rightDepth;
    }
}

小结：如果知道递归法的话, 递归法确实比迭代法简洁很多

559. N 叉树的最大深度(Easy)

题目链接:559. N 叉树的最大深度

题目：给定一个 N 叉树，找到其最大深度。
最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。
N 叉树输入按层序遍历序列化表示，每组子节点由空值分隔（请参见示例）。

示例 1：
输入：root = [1,null,3,2,4,null,5,6]
输出：3

思路：如上
时间复杂度：O(n) 空间复杂度：O(log n)

解决方案：

class Solution {
    public int maxDepth(Node root) {
        if (root == null) return 0;
        int max = 0;
        for(Node n : root.children) {
            int branchDepth = maxDepth(n);
            if (max < branchDepth) {
                max = branchDepth;
            }
        }
        return max + 1;
    }
}

小结：无

111. 二叉树的最小深度(Easy)

题目链接:111. 二叉树的最小深度

题目：给定一个二叉树，找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明：叶子节点是指没有子节点的节点。

示例 1：
输入：root = [3,9,20,null,null,15,7]
输出：2

示例 2：
输入：root = [2,null,3,null,4,null,5,null,6]
输出：5

思路：这题目在第十四天的练习中有迭代法, 这里使用递归法去计算.
时间复杂度：O(n) 空间复杂度：O(log n)

解决方案：

class Solution {
    public int minDepth(TreeNode root) {
        if (root == null) return 0;
        if(root.left == null && root.right == null) return 1;
        
        int leftDepth = minDepth(root.left) + 1;
        int rightDepth = minDepth(root.right) + 1;
        if(root.left == null) return rightDepth;
        if(root.right == null) return leftDepth;
        return leftDepth < rightDepth ? leftDepth : rightDepth;
    }
}

小结：
注意如下错误:

222. 完全二叉树的节点个数(Middle)

题目链接:222. 完全二叉树的节点个数

题目：给你一棵 完全二叉树 的根节点 root ，求出该树的节点个数。
示例 1：
输入：root = [1,2,3,4,5,6]
输出：6

示例 2：
输入：root = []
输出：0

示例 3：
输入：root = [1]
输出：1

思路：迭代法套用模板可以很快写出题解, 但是这里我们使用递归, 使用后序遍历可以很快解决这种先子树再中间的计数问题.
时间复杂度：O(n) 空间复杂度：O(log n)

解决方案：

class Solution {
    public int countNodes(TreeNode root) {
        if (root == null) return 0;
        int leftSum = countNodes(root.left);
        int rightSum = countNodes(root.right);
        return leftSum + rightSum + 1;
    }
}

小结：后序遍历可以很快解决这种先子树再中间的计数问题

---

总结

二叉树的递归有一定的套路, 首先要确定好是要进行哪种遍历.