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

本次全部使用递归的方式实现

104.二叉树的最大深度

递归法，后序遍历

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0
        left_depth = self.maxDepth(root.left)
        right_depth = self.maxDepth(root.right)
        return 1 + max(left_depth, right_depth)

111.二叉树的最小深度

注意：最小深度是从根节点到最近叶子节点的最短路径上的节点数量。注意是叶子节点，左右孩子都为空的节点才是叶子节点。因此终止条件多了几个判断。

class Solution:
    def minDepth(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0
        left_depth = self.minDepth(root.left)
        right_depth = self.minDepth(root.right)
        if not root.left:
            return 1 + right_depth
        if not root.right:
            return 1 + left_depth

        return 1 + min(left_depth, right_depth)

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

普通二叉树的递归方式

class Solution:
    def countNodes(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0
        left_nodes = self.countNodes(root.left)
        right_nodes = self.countNodes(root.right)
        return 1 + left_nodes + right_nodes

完全二叉树的递归方式

class Solution:
    def countNodes(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0
        left = root.left
        right = root.right
        left_depth = 0
        right_depth = 0
        while left:
            left = left.left
            left_depth += 1
        while right:
            right = right.right
            right_depth += 1
        if left_depth == right_depth:
            return (2 << left_depth) - 1

        return 1 + self.countNodes(root.left) + self.countNodes(root.right)