Leetcode 104. 二叉树的最大深度

题目描述

给定一个二叉树，找出其最大深度。

二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

说明: 叶子节点是指没有子节点的节点。

递归解法

二叉树根节点为空，则树高度为 0，根节点不为空，则高度为左子树、右子树的最大值加一。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        return max(self.maxDepth(root.left),self.maxDepth(root.right))+1

迭代解法

以队列保存每层的节点，层次遍历，每加一层高度加一。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def maxDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        arr,depth=[root],0
        while len(arr)>0:
            depth+=1
            for i in arr[:]:
                node=arr.pop(0)
                if node.left:
                    arr.append(node.left)
                if node.right:
                    arr.append(node.right)
        return depth