二叉树最大深度python

题目链接：104. 二叉树的最大深度 - 力扣（Leetcode）

栈 + 标识 方法能解决大多数二叉树遍历问题

栈元素： （所在深度，本身结点）

当栈不空，取出栈顶元素，判断其是不是叶子结点，如果是把这个叶子结点的深度放到res列表中，如果不是叶子节点，把他的子结点放入栈，子结点的深度就是当前处理结点的深度+1。

栈空证明每个结点都被处理了。其中所有叶子结点的深度都放入res了，所有叶子结点的深度中的最大值就是这棵二叉树的最大深度。

小提一句：一开始的时候我犯了一个简答的编写错误：

 # 如果是叶子结点则把当前深度放到res
            if not node.right and not node.left:
                res.append(dep_now)
            # 非叶子结点则把他的子结点和所在的深度入栈
            elif node.right:
                stack.append((dep_now + 1, node.right))
            elif node.left:
                stack.append((dep_now + 1, node.left))

if 下面有多个elif 遇到一个符合的elif就结束了，不会再去看其他的elif。题目中不是叶子结点的话，左右结点一旦非空就都得进栈，所以左右结点都得检查。

正确代码如下：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def maxDepth(self, root: Optional[TreeNode]) -> int:
        if not root : return 0
        stack = [(1,root)]
        res = [1]
        

        while stack:
            
           
            dep_now,node = stack.pop()
            # 如果是叶子结点则把当前深度放到res
            if not node.right and not node.left:
                res.append(dep_now)
            # 非叶子结点则把他的子结点和所在的深度入栈
            else:
                if node.right:
                    stack.append((dep_now+1,node.right))
                if node.left:
                    stack.append((dep_now+1,node.left))

        # 结束循环时所有结点已经遍历，所有叶子结点的深度都放入了res，res中最大的数就是最大深度
        return max(res)