给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
方法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
stack = [root]
n = 0
while stack:
next_layer = []
for root in stack:
# 在这个循环中会将root的左右子树入栈,放在next_layer中,所以stack中存放的都是同一层的树
# stack中有多少个元素就说明该层有多少个节点,就要将节点的值放进temp中。
if root.left:
next_layer.append(root.left)
if root.right:
next_layer.append(root.right)
n+=1
stack = next_layer
return n
方法2:递归
思路:找最大深度也就是找左子树和右子树中各自最大深度中最大的然后加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
return max(self.maxDepth(root.left)+1,self.maxDepth(root.right)+1)