题目描述:
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
解题:
- 方法一:递归
这是一道简单题,解法也比较多,递归的思路很好理解,树的深度等于左子树和右子树中深度较大的那个深度加1。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if not root:
return 0
return max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1
- 方法二:层次遍历
树的层序遍历 / 广度优先搜索往往利用 队列 实现。
关键点: 每遍历一层,则计数器 +1+1 ,直到遍历完成,则可得到树的深度。
算法解析:
- 特例处理: 当 root 为空,直接返回 深度 00 。 初始化: 队列 queue (加入根节点 root ),计数器 res =0。
- 循环遍历: 当 queue 为空时跳出。 初始化一个空列表 tmp ,用于临时存储下一层节点;
- 遍历队列: 遍历 queue中的各节点 node ,并将其左子节点和右子节点加入 tmp;
- 更新队列: 执行 queue = tmp ,将下一层节点赋值给queue; 统计层数: 执行 res += 1 ,代表层数加 11; 返回值: 返回 res 即可。
class Solution(object):
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if not root:
return 0
queue = [root]
res = 0
while queue:
tmp = []
for node in queue:
if node.left:tmp.append(node.left)
if node.right:tmp.append(node.right)
queue = tmp
res += 1
return res