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

文档讲解：代码随想录

视频讲解：代码随想录B站账号

状态：看了视频题解和文章解析后做出来了

104.二叉树的最大深度

1. 递归

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0

        result = self.depth(root)

        return result

    def depth(self, root) -> int:
        if not root:
            return 0

        left_height = self.depth(root.left)
        right_height = self.depth(root.right)

        return 1 + max(left_height, right_height)

思路：

• 前序遍历：得到的是深度，因为中节点优先处理，0->1->2每次遍历的是当前层的深度
• 后序遍历：得到的是高度，因为子节点的高度返回给母节点，所以是到根节点是高度

递归第一步：确定参数和返回值，因为我们需要深度/高度所以肯定返回整数。

递归第二步：确定终止条件，当节点为空，返回0，也就是空节点的深度

递归第三步：确定每一层的处理逻辑，递归左节点和右节点，最后返回max+1。思考一下如果是空节点，那就会返回给母节点1.

2. 层序遍历

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0

        que = deque([root])
        result = 0

        while que:
            for i in range(len(que)):
                cur = que.popleft()
                if cur.left:
                    que.append(cur.left)
                if cur.right:
                    que.append(cur.right)
            result += 1

        return result

思路：依然觉得层序遍历是最直接的方式，直接遍历每层，遍历完一层result += 1就完事了。

111.二叉树的最小深度

class Solution:
    def minDepth(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0

        result = 0
        result = self.depth(root)

        return result


    def depth(self, root):
        if not root:
            return 0

        left_height = self.depth(root.left)
        right_height = self.depth(root.right)

        if root.left and not root.right:
            return 1 + left_height
        elif root.right and not root.left:
            return 1 + right_height

        return 1 + min(left_height, right_height)

细节：这道题和最大深度相似，使用后序遍历。但有一个小坑，就是在返回值的时候，首先要判断左子树和右子树是否为空，如果为空就不能返回它们的深度，因为这样取min就一定是0了。题目要求的是根节点到叶节点的最小深度。

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

递归

class Solution:
    def countNodes(self, root: Optional[TreeNode]) -> int:
        return self.count(root)

    def count(self, root):
        if not root:
            return 0

        left_count = self.count(root.left)
        right_count = self.count(root.right)

        return left_count + right_count + 1

层序遍历

class Solution:
    def countNodes(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0

        que = deque([root])
        count = 1
        while que:
            for i in range(len(que)):
                cur = que.popleft()
                if cur.left:
                    que.append(cur.left)
                if cur.right:
                    que.append(cur.right)
            count += len(que)
            
        return count

思路：让我们喊出那句话：层序遍历永远的神。