leetcode104 求二叉树的最大深度——深度遍历DFS/层序遍历BFS

直接上车

① DFS 前序

这里其实先不用搞清楚“前序” 的特点，你只要知道DFS是一道一道从根节点到树叶节点的路径遍历就可以了，思路是我们专门搞一个存子树深度的变量res，如果当前的遍历路径的深度比这个res更大，那这个深度就替换res变成暂时的最大深度，直到DFS结束，最后的res的值最大深度。

这里还是要注意 一下

nonlocal res

关于Python global、local、nonlocal的作用，之后专门写一篇文章。

class Solution:

    def maxDepth(self, root: TreeNode) -> int:
        if not root:
            return 0
        
        res = 0
        def dfs(root, depth):
            if not root:
                nonlocal res
                res = max(res, depth)
                return
            
            dfs(root.left, depth+1)
            dfs(root.right, depth+1)

        dfs(root, 0)
        return res

② DFS 后序

我们知道，①是遍历完根节点到树叶节点才进行一次max()的比较。②和①区别的地方是，②是“一边遍历一边比较”。所以②每一个节点都有一个返回值，这个返回值代表着它的子树的最大深度。
说起来还是很抽象，那就画图解决，没有办法，我的脑子就是很fw。

这个图就从底部红色的叶子节点开始看，他们实际上不存在，是一个NULL。所以会走到代码的base case，返回一个0的值，对于真正的叶子节点，他们实际上左右两边都返回0，所以给父节点返回的就是max(0,0)+1。

def maxDepth(self, root: TreeNode) -> int:
        if not root:
            return 0

        def dfs(root):
            if not root:
                return 0

            left_depth = dfs(root.left)
            right_depth = dfs(root.right)

            return max(left_depth, right_depth) + 1

        return dfs(root)

③层序遍历BFS

BFS比较简单，我的fw脑子也想得明白，你也一定可以

class Solution(object):
    def maxDepth(self, root):
       
        if not root:
            return 0
        queue = collections.deque()
        queue.append(root)
        res = 0
        while queue:
            res += 1
            for i in range(len(queue)):
                node = queue.popleft()
                if node.left: queue.append(node.left)
                if node.right: queue.append(node.right)
        return res