题目
Given a binary tree, find its maximum depth.
The maximum depth is the number of nodes along the longest path from the root node down to the farthest leaf node.
Note: A leaf is a node with no children.
Example:
Given binary tree [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return its depth = 3.
总结
此题首先想到的是肯定需要遍历所有节点,但是由于很久没接触二叉树的题目,没有更进一步想到常用的两种遍历方法:深度优先遍历(前中后序)和广度优先遍历(层次)。
我最初的解法形式上是使用深度优先遍历,但是代码很不美观。没有很好的保存maxdepth
初始代码:
class Solution:
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
maxd = 0
if root is not None:
maxd = self.depth(root, 1, maxd)
return maxd
def depth(self, node, d, maxd):
if d > maxd:
maxd = d
if node.left is not None:
maxd = self.depth(node.left, d + 1, maxd)
if node.right is not None:
maxd = self.depth(node.right, d + 1, maxd)
return maxd
可以考虑把maxd作为全局变量
还有一种更elegant的方法:
class Solution:
def maxDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if root is None:
return 0
return 1 + max(self.maxDepth(root.left), self.maxDepth(root.right))
这种递归的方法相当于反过来计算depth,最底层是0,向上依次加一,解决了保存maxd的难题
还有一种方法是使用广度优先遍历,每向下一层maxd + 1,不过要用到queue支持。