DFS-LeetCode104. 二叉树的最大深度

1、题目描述

https://leetcode-cn.com/problems/maximum-depth-of-binary-tree/

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

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

说明: 叶子节点是指没有子节点的节点。

2、代码详解

递归写法（推荐）

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):
    # 递归写法
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        # 递归法
        if root is None:
            return 0
        else:
            left_height = self.maxDepth(root.left)
            right_height = self.maxDepth(root.right)
            return max(left_height, right_height) + 1

时间复杂度：我们每个结点只访问一次，因此时间复杂度为 O(N)， 其中 N 是结点的数量。

空间复杂度：在最糟糕的情况下，树是完全不平衡的，例如每个结点只剩下左子结点，递归将会被调用 N 次（树的高度），因此保持调用栈的存储将是 O(N)。但在最好的情况下（树是完全平衡的），树的高度将是 log(N)。因此，在这种情况下的空间复杂度将是 O(log(N))。

非递归写法（利用栈）

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):   

    # 非递归法：在栈的帮助下将上面的递归转换为迭代
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        # 使用 DFS 策略访问每个结点，同时在每次访问时更新最大深度
        stack = []  # （depth，node）
        if root is not None:
            stack.append((1, root))

        depth = 0
        while stack != []:
            current_depth, root = stack.pop()
            if root is not None:
                depth = max(depth, current_depth)
                stack.append((current_depth + 1, root.left))
                stack.append((current_depth + 1, root.right))

        return depth

时间O(N)、空间O(N)