104_二叉树的最大深度

"""
给定一个二叉树，找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

说明: 叶子节点是指没有子节点的节点。
示例：
给定二叉树 [3,9,20,null,null,15,7]，

    3
   / \
  9  20
    /  \
   15   7
返回它的最大深度 3 。
"""


# 递归
def maxDepth(root):

    if root is None:
        return 0
    else:
        left_height = maxDepth(root.left)
        right_height = maxDepth(root.right)
        return max(left_height, right_height)+1

# 迭代
def maxDepth2(root):
    stack = []
    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

print(maxDepth(node1))