1、递归实现,首先是自己写的一种递归实现,感觉比较直观的利用的当前节点和当前节点的深度(一种“自顶向下”的广度优先搜索)
class Solution(object):
def maxDepth(self, root: TreeNode):
def depth_func(node, node_d):
if node is None:
return node_d
return max(depth_func(node.left, node_d + 1), depth_func(node.right, node_d + 1))
if root is None:
return 0
max_d = max(depth_func(root.left, 1), depth_func(root.right, 1))
return max_d
2、官方给出的参考答案
一种“自底向上”的广度优先搜索,如果我们知道了左子树和右子树的最大深度 l 和 r,那么该二叉树的最大深度即为
max(l,r) +1
class Solution(object):
def maxDepth(self, root: TreeNode):
if root is None:
return 0
return max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1
3、使用 二叉树的层序遍历 获取二叉树的最大深度
class Solution(object):
def maxDepth(self, root: TreeNode):
if root is None:
return 0
depth = 0
queue = [root]
while queue:
current_level_size = len(queue)
for i in range(current_level_size):
node = queue.pop(0)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
depth += 1
return depth