LeetCode 热题100-37 二叉树的最大深度

二叉树的最大深度

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

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

示例 1:

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

示例 2:

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

提示:

  • 树中节点的数量在 [0, 104] 区间内。
  • -100 <= Node.val <= 100

以前本科数据结构做过,感觉递归还是巧妙啊

class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if root == None:return 0
        high = 0
        def dfs(root):
            if root == None:return 0
            if root.left ==None and root.right == None:return 1
            return max(dfs(root.left),dfs(root.right))+1
        return dfs(root)

然后看了看k神的题解,发现考虑是否有左右孩子节点的那个if没有必要...

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if root == None:return 0
        if root.left==None and root.right==None:return 1
        return max(self.maxDepth(root.left),self.maxDepth(root.right))+1

而且也不需要再内置一个函数了...