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
}

---

0x03 我的小作品

欢迎体验我的作品之一：小编辑器-XCompiler
小巧的在线编辑器~
App Store 搜索即可~

---