今天我们来做一道LeetCode上的题目，原题链接：剑指 Offer 55 - I. 二叉树的深度
题目描述
- 输入一棵二叉树的根节点，求该树的深度。从根节点到叶节点依次经过的节点（含根、叶节点）形成树的一条路径，最长路径的长度为树的深度。
- 例如：
给定二叉树 [3,9,20,null,null,15,7]，
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
- 提示：
节点总数 <= 10000
思路分析
- 递归实现深度优先
- 队列实现广度优先
代码
# Python
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
# # DFS
# if not root: return 0
# return max(self.maxDepth(root.right), self.maxDepth(root.left)) + 1
# BFS
if not root: return 0
queue, res = [root], 0
while queue:
tmp = []
for node in queue:
if node.left: tmp.append(node.left)
if node.right: tmp.append(node.right)
res += 1
queue = tmp
return res
总结
- 深度优先代码更简洁
- 广度优先逻辑更直观
附录
本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情
