http://www.lintcode.com/zh-cn/problem/maximum-depth-of-binary-tree/
http://www.jiuzhang.com/solutions/maximum-depth-of-binary-tree/
描述:
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的距离。
样例
给出一棵如下的二叉树:
1
/ \
2 3
/ \
4 5
这个二叉树的最大深度为3.
标答代码不是很容易理解,我以我的代码作解释。
我的代码虽然啰嗦了一些,也不是最简洁最好的写法,但是对于理解起来比较好。
这道题的解法是递归遍历所有的节点,找到一个深度最大值。
具体方法是在每一个节点,都去递归遍历左子树和右子树,并在其中找一个最大值。
"""
Definition of TreeNode:
class TreeNode:
def __init__(self, val):
self.val = val
self.left, self.right = None, None
"""
class Solution:
def dfs(self, node, depth):
if node.left == None and node.right == None: # 如果已经到了叶子节点,那么返回当前深度
return depth
max = 0
if node.left:
max = self.dfs(node.left, depth + 1) # 不然接着往下遍历
if node.right:
dep = self.dfs(node.right, depth + 1) # 在左子树和右子树中找一个深度最大值
if dep > max:
return dep
return max # 返回深度最大值
"""
@param root: The root of binary tree.
@return: An integer
"""
def maxDepth(self, root):
max = 1
if root == None: # 惯例判断是否为空
return 0
if root.left:
max = self.dfs(root.left, 2)
if root.right:
dep = self.dfs(root.right, 2)
if dep > max:
return dep
return max