【Python刷题】二叉树的最大深度

问题描述

给定一个二叉树 root ，返回其最大深度。

二叉树的 最大深度 是指从根节点到最远叶子节点的最长路径上的节点数。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：3

示例 2：

输入：root = [1,null,2]
输出：2

解决办法

• 定义一个深度遍历的函数，如果不存在根结点，直接返回结果（0）；
• 如果存在，则开始遍历根结点的左子树和右子树，返回两者中的最大值。注意这是需要给深度参数+1；
• 最后调用这个函数，并返回结果。

代码示例

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