给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回它的最大深度 3 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-depth-of-binary-tree
python
# 0104.二叉树的最大深度
class Solution:
def levelOrder(self, root: TreeNode) -> int:
"""
迭代法:双端队列,每次把单层的节点遍历出队列,另外将对应的左右节点加入队列
:param root:
:return:
"""
results = []
if not root:
return 0
from collections import deque
queue = deque([root]) # 初始化队列
while queue:
size = len(queue) # 遍历队列单层长度
res = [] # 每次遍历时初始化加入结果集中的列表
for _ in range(size):
cur = queue.popleft()
res.append(cur.val) # 加入结果集中
if cur.left: # 添加当前pop节点的左节点进入队列
queue.append(cur.left)
if cur.right: # 添加当前pop节点的右节点进入队列
queue.append(cur.right)
results.append(res) # 当层中的所有节点的值放入list中
return len(results)
golang
package binaryTree
import (
"container/list"
)
// 迭代遍历
func maxDepth(root *TreeNode) int {
var res = [][]int{}
if root == nil { // 空时返回
return 0
}
queue := list.New() // 队列初始化
queue.PushBack(root)
for queue.Len() > 0 {
length := queue.Len()
tmpArr := []int{}
for i:=0;i<length;i++ { // 遍历当层的节点
node := queue.Remove(queue.Front()).(*TreeNode) // 当次节点
if node.Left != nil { // 节点的左节点入队
queue.PushBack(node.Left)
}
if node.Right != nil { // 节点的右节点入队
queue.PushBack(node.Right)
}
tmpArr = append(tmpArr, node.Val) // 节点值加入结果集
}
res = append(res, tmpArr)
}
return len(res)
}