题解 | #二叉树的最大深度#

二叉树的最大深度

https://www.nowcoder.com/practice/8a2b2bf6c19b4f23a9bdb9b233eefa73

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 
# @param root TreeNode类 
# @return int整型
#
class Solution:
    def maxDepth(self , root: TreeNode) -> int:
        # write code here
        if root == None:
            return 0

        l = [root] # 当前列表,存储遍历的当前节点以及下级子节点
        res = []#定义数组,每个元素则为每层的元素

        while l:
            sub = []  #每层的节点
            for i in range(len(l)): #遍历当前列表
                cur = l.pop(0) #弹出节点
                sub.append(cur.val) #压入每层子列表
                if cur.left: #如果有左子树,则压入总列表
                    l.append(cur.left)
                if cur.right: #如果有右子树,则压入总列表
                    l.append(cur.right)
            res.append(sub) #压入结果列表

        return len(res) #计算结果列表的元素个数,即为二叉树深度

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务