263 字
1 分钟
55-I.二叉树的深度
一、 题目
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3 / \ 9 20 / \ 15 7返回它的最大深度 3 。
提示:
节点总数 <= 10000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/er-cha-shu-de-shen-du-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、 解法
2.1 递归
func maxDepth(_ root: TreeNode?) -> Int { guard let tree = root else { return 0 } if tree.left == nil, tree.right == nil { return 1 } return max(maxDepth(tree.left), maxDepth(tree.right)) + 1}2.2 非递归
func maxDepth(_ root: TreeNode?) -> Int { var deep = 0 guard let root = root else { return deep } var queue: [TreeNode] = [root] var currentLevelCount = 1 var nextLevelCount = 0
while queue.isEmpty == false { let current = queue.removeFirst() currentLevelCount -= 1
if let left = current.left { queue.append(left) nextLevelCount += 1 }
if let right = current.right { queue.append(right) nextLevelCount += 1 }
if currentLevelCount == 0 { deep += 1 currentLevelCount = nextLevelCount nextLevelCount = 0 } }
return deep}