0x00 题目
给定一个二叉树,找出其最大深度
。
二叉树的深度为根节点到最远叶子节点
的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
0x01 思路
方式一:
通过递归左右子树,然后取最大值即可。
方式二:
通过二叉树的层序遍历方式,从当前节点,依次向下寻找。
每往下一层,同时记录深度。
直到最后一层为空时,则找到了最远的叶子节点。
0x02 解法
语言:Swift
树节点:TreeNode
public class TreeNode {
public var val: Int
public var left: TreeNode?
public var right: TreeNode?
public init() { self.val = 0; self.left = nil; self.right = nil; }
public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
self.val = val
self.left = left
self.right = right
}
}
递归方式:
func maxDepth(_ root: TreeNode?) -> Int {
if root == nil {
return 0
}
return max(maxDepth(root?.left), maxDepth(root?.right)) + 1
}
迭代方式:
func maxDepth_(_ root: TreeNode?) -> Int {
guard let root = root else { return 0 }
// 深度,因为当 tmp 最后为空时,depth 依旧加了 1,所以初始化为 0
var depth = 0
var queue: [TreeNode] = []
queue.append(root)
while !queue.isEmpty {
var tmp: [TreeNode] = []
for node in queue {
if let left = node.left {
tmp.append(left)
}
if let right = node.right {
tmp.append(right)
}
}
queue = tmp
depth += 1
}
return depth
}
挑战 1 天学会五笔的最高纪录
你可行?