给定一个二叉树,根节点为root,对于两个节点p和q,返回它们最近的公共祖先节点。
样例:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
输出:5
解析: 这个问题可以通过后序遍历二叉树来解决。首先,我们需要找到节点p和q的最近公共祖先节点的左子树和右子树。然后,我们比较这两个子树是否包含根节点。如果其中一个子树包含根节点,那么根节点就是最近的公共祖先节点。否则,我们就在另一个子树中继续寻找。
代码如下:
class Solution:
def lowestCommonAncestor(self, root, p, q):
def getPath(root, target):
path = []
while root:
path.append(root)
if root == target:
return path
root = root.left if target.val < root.val else root.right
return None
path_p = getPath(root, p)
path_q = getPath(root, q)
ancestor = None
for u, v in zip(path_p, path_q):
if u and v:
ancestor = u
else:
break
return ancestor.val if ancestor else None
在这段代码中,我们首先定义了一个辅助函数getPath,用于获取从根节点到目标节点的路径。然后,我们分别获取了从根节点到节点p和节点q的路径。最后,我们比较这两个路径,找到最后一个公共节点,即为最近的公共祖先节点