题目:二叉树的最近公共祖先
题解:
1) 首先想到用迭代法表示后序遍历,因为寻找公共祖先应该从下往上
2)迭代法考虑终止条件:如果寻找到其中一个节点,那么就返回这个节点;
3)考虑几种情况:作为中间节点,当左右节点都找到时,它自己本身就是最近公共祖先;如果左/右面找到了(只有一个),那么它也只能继续向上回溯;如果都没找到就向上一级返回空。
4)注意一个额外情况:当要寻找的节点其中一个是公共祖先的时候,这种情况也包含在了上述情况中。因为如果找到第二个节点,根据终止条件也会返回第二个节点。
代码:
class Solution(object):
def lowestCommonAncestor(self, root, p, q):
"""
:type root: TreeNode
:type p: TreeNode
:type q: TreeNode
:rtype: TreeNode
"""
if root == p or root == q:
return root
if not root:
return None
left = self.lowestCommonAncestor(root.left, p, q)
right = self.lowestCommonAncestor(root.right, p, q)
if left != None and right != None:
return root
elif left == None and right != None:
return right
else:
return left