LeetCode104 | 二叉树的最大深度

读前福利：几百本互联网技术书籍送给你们

天天分享一个LeetCode题目

天天 5 分钟，一块儿进步

LeetCode104 二叉树的最大深度，地址: leetcode-cn.com/problems/ma…

树结点定义

class TreeNode(object):
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
复制代码

递归方式

递归方式比较简单，就是一个后续遍历，而后在遍历到根结点的时候进行左右比较，选择大的孩子

def maxDepth_dfs(self, root):
    if not root:
        return 0
    else:
        max_left = self.maxDepth_dfs(root.left)
        max_right = self.maxDepth_dfs(root.right)
        return max(max_left, max_right) + 1
复制代码

看，是否是一个很简单的后续遍历，就是最后选择大的分支

可是非递归的方式对于不少从上到下的遍历也很适用，并且简单

非递归方式

和

【此处是层次遍历的连接】

的思路一致，就是一个层序遍历的过程

初始化一个 node_depth = 0，而后遍历到每一层的时候进行 node_depth + 1，最后返回 node_depth

看代码

def maxDepth_bfs(self, root):
    if not root:
        return 0
    queue = collections.deque()
    # 初始化深度为 0
    node_depth = 0
    # 初始化队列中的结点元素 root
    queue.appendleft(root)
    while queue:
        # 每一层的遍历，深度 +1
        node_depth += 1
        # 记录每一层的结点集合
        tmp_queue = []
        for node in queue:
            if node.left:
                tmp_queue.append(node.left)
            if node.right:
                tmp_queue.append(node.right)
        queue = tmp_queue

    return node_depth
复制代码

尤为是不少地方都会使用像这种 BFS 的方式进行遍历访问，并且主要适用于一些从顶向下的题目

所有代码

# -*- coding:utf-8 -*-
# !/usr/bin/env python

import collections

# 树结点类
class TreeNode(object):
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


class Solution(object):
    def maxDepth_dfs(self, root):
        if not root:
            return 0
        else:
            max_left = self.maxDepth_dfs(root.left)
            max_right = self.maxDepth_dfs(root.right)
            return max(max_left, max_right) + 1

    def maxDepth_bfs(self, root):
        if not root:
            return 0
        queue = collections.deque()
        # 初始化深度为 0
        node_depth = 0
        # 初始化队列中的结点元素 root
        queue.appendleft(root)
        while queue:
            # 每一层的遍历，深度 +1
            node_depth += 1
            # 记录每一层的结点集合
            tmp_queue = []
            for node in queue:
                if node.left:
                    tmp_queue.append(node.left)
                if node.right:
                    tmp_queue.append(node.right)
            queue = tmp_queue

        return node_depth


if __name__ == "__main__":
    # 新建节点
    root = TreeNode('A')
    node_B = TreeNode('B')
    node_C = TreeNode('C')
    node_D = TreeNode('D')
    node_E = TreeNode('E')
    node_F = TreeNode('F')
    node_G = TreeNode('G')
    node_H = TreeNode('H')
    node_I = TreeNode('I')
    node_J = TreeNode('J')
    # 构建二叉树
    # A
    # / \
    # B C
    # / \ / \
    # D E F G
    # / \
    # H I
    # \
    # J
    root.left, root.right = node_B, node_C
    node_B.left, node_B.right = node_D, node_E
    node_C.left, node_C.right = node_F, node_G
    node_D.left, node_D.right = node_H, node_I
    node_I.right = node_J

    s = Solution()
    print(s.maxDepth_dfs(root))
    print(s.maxDepth_bfs(root))
复制代码

Github

github.com/xiaozhutec/…