LeetCode-剑指 Offer 55 - I-二叉树的深度

剑指 Offer 55 - I. 二叉树的深度

题目说明

输入一棵二叉树的根节点，求该树的深度。从根节点到叶节点依次经过的节点（含根、叶节点）形成树的一条路径，最长路径的长度为树的深度。

示例

给定二叉树 [3,9,20,null,null,15,7]，

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

题解思路

1、广度优先搜索：按层遍历，每遍历一层，计数值+1
2、深度优先搜索：树的深度=左右子树深度的最大值 + 1

代码实现

1、广度优先搜索
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

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



2、深度优先搜索
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        if not root:return 0
        ans = max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1
        return ans