【LeetCode】Python代码实现 之 面试题55 - I. 二叉树的深度

输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。

例如:

给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。

解答:
树的遍历的方法可以分为:深度优先搜索(DFS),广度优先搜索(BFS)
DFS包括:先序遍历,中序遍历,后序遍历
BFS包括:层序遍历
解法1:DFS,树的深度为左子树和右子树深度的最大值加1.
解法2:BFS,每遍历一层,计数器加1,直到遍历完成,得到树的深度

解法1代码:

# 使用递归
from binarytree import build

def max_depth(root):
    if not root: return 0
    return max(max_depth(root.left), max_depth(root.right))+1

values = [3,9,20,None,None,15,7]
root = build(values)
max_depth(root)

解法2代码:
算法解析:
1 特例处理: 当 root​ 为空,直接返回 深度 0 。
2 初始化: 队列 queue (加入根节点 root ),计数器 res = 0。
3 循环遍历: 当 queue 为空时跳出。
初始化一个空列表 tmp ,用于临时存储下一层节点;
遍历队列: 遍历 queue 中的各节点 node ,并将其左子节点和右子节点加入 tmp;
更新队列: 执行 queue = tmp ,将下一层节点赋值给 queue;
统计层数: 执行 res += 1 ,代表层数加 1;

from binarytree import build

def max_depth(root):
	if not root: return 0
	queue, tmp = [root], 0
	res = 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

values = [3,9,20,None,None,15,7]
root = build(values)
max_depth(root)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值