题目:
思路+代码:
层序遍历:层序遍历计算二叉树层数,时间复杂度O(n),空间复杂度O(n)
class Solution:
def maxDepth(self, root: TreeNode) -> int:
# # 层序遍历计算二叉树层数,时间复杂度O(n),空间复杂度O(n)
if not root:
return 0
depth = 0
node_list = []
node_list.append(root)
while node_list:
layer_list = []
while node_list:
node = node_list.pop(0)
if node.left:
layer_list.append(node.left)
if node.right:
layer_list.append(node.right)
depth += 1
node_list.extend(layer_list)
return depth
后序遍历递归返回二叉树深度
# 递归结束条件:当子树根节点不存在时,返回0
# 递推工作:在当前子树,深度= max(左子树深度,右子树深度) + 1
# 返回值:当前树深度
class Solution:
def maxDepth(self, root: TreeNode) -> int:
# 后序遍历递归返回二叉树深度
# 递归结束条件:当子树根节点不存在时,返回0
# 递推工作:在当前子树,深度= max(左子树深度,右子树深度) + 1
# 返回值:当前树深度
if not root:
return 0
return max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1