此题的意思可以概括为有一堆最深的叶子节点,这些叶子节点的公共子节点是谁。可以看到在递归的过程中需要记录一下树的深度。因此新建一个函数,返回值不光有结果,还有对应的子树深度。首先分析几个特殊情况,很简单不再赘述。对于一般情况则是得到两个子树的高度,如果其中一个子树比另一个高,那么只需要返回高的那个子树的结果,因为我们只想要最深的叶子节点的公共父亲。如果两个子树的高度一样高,说明这两个子树都有着最深的叶子节点,为了能全部包含,返回当前的root.
代码如下:
# 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 depthandtree(self, root):
if root == None:
return root, 0
if root.left == None and root.right == None:
return root, 1
if root.left != None and root.right == None:
tree, depth = self.depthandtree(root.left)
return tree, depth + 1
if root.left == None and root.right != None:
tree, depth = self.depthandtree(root.right)
return tree, depth + 1
left_tree, left_depth = self.depthandtree(root.left)
right_tree, right_depth = self.depthandtree(root.right)
if left_depth > right_depth:
return left_tree, left_depth+1
elif right_depth > left_depth:
return right_tree, right_depth+1
else:
return root, left_depth + 1
def lcaDeepestLeaves(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
result, _ = self.depthandtree(root)
return result