剑指 Offer 55 - I. 二叉树的深度
题目说明
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
示例
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
题解思路
1、广度优先搜索:按层遍历,每遍历一层,计数值+1
2、深度优先搜索:树的深度=左右子树深度的最大值 + 1
代码实现
1、广度优先搜索
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if not root:return 0
queue = collections.deque()
queue.append(root)
ans = 0
while queue:
n = len(queue)
for i in range(n):
node = queue.popleft()
left, right = node.left, node.right
if left:
queue.append(left)
if right:
queue.append(right)
ans += 1
return ans
2、深度优先搜索
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if not root:return 0
ans = max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1
return ans