swift算法：二叉树的最大深度

1、描述

给定一个二叉树，找出其最大深度。

二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

说明：叶子节点是指没有子节点的节点

例1：给定二叉树 [3, 9, 20, nil, nil, 15, 7]

         

        返回它的最大深度 3

例1：给定二叉树 [1, 2, 7, 3, nil, nil,  4, 5, 6]

         

        返回它的最大深度 4

2、算法：

1）二叉树节点

public class TreeNode{
    public var val : Int
    public var left : TreeNode?
    public var right :TreeNode?
    public init(_ val : Int){
        self.val = val
        self.left = nil
        self.right = nil
    }
}

2）求最大深度

解法一：递归

思想：深度优先搜索DFS

时间复杂度：O(n)

举例 [[1, 2, 7, 3, nil, nil,  4, 5, 6]] 图示：

func maxDepth(_ root: TreeNode?) -> Int {
        /*
         递归：深度优先搜索DFS
         */
        guard root != nil else{
            return 0
        }
        let left_height = maxDepth(root?.left)
        let right_height = maxDepth(root?.right)
        return max(left_height, right_height)+1
    }
    

解法二：迭代

思想：利用栈将递归转换为迭代，使用dfs策略访问每个节点，同时在每次访问时更新最大深度
           从包含根结点且相应深度为 1 的栈开始。然后我们继续迭代：将当前结点弹出栈并推入子结点。每一步都会更新深度。

时间复杂度：O(n)

func maxDepth(_ root: TreeNode?) -> Int {
        /*
         迭代：
         */
        var stack : [(TreeNode?,Int)] = []
        if root != nil {
            stack.append((root,1))
        }else{
            return 0
        }
        var depth = 1
        while !stack.isEmpty {
            //取出根节点
            let current = stack.removeFirst()
            let root : TreeNode? = current.0
            let current_depth = current.1
            if root != nil{
                depth = max(depth, current_depth)
                stack.append((root?.left, current_depth+1))
                stack.append((root?.right, current_depth+1))
            }
        }
        return depth
    }