当前位置:   article > 正文

LeetCode 236. 二叉树的最近公共祖先——Python实现(递归,哈希表)

LeetCode 236. 二叉树的最近公共祖先——Python实现(递归,哈希表)

1、直接用递归的方法

     使用递归的方法,找出二叉树中两个节点的最近公共祖先,分析如下:

      对于两个节点p和q的公共祖先r,他们要么在r的同一边,要么在这个节点的两边,因此,只要满足这两个条件即可,也就是说: 只要 f(r.child, p) and f(r.child,q) 成立,这个r就可能是它俩的祖先。换一个角度说,r有两条支线,现在不管p和 q在哪条支线,只要两条支线都为真(即包含目标节点),并且有一个交集,那这个交集这就是公共祖先。 这样的话,我们只要通过 p 和 q 一路标记他们经过的路径,然后如果两个标记在某一个节点相遇了,那这样就可以找出最近公共祖先了。

  1. class Solution:
  2. def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
  3. # 如果已经访问到根节点都没找到相应的目标节点,那么说明这条线应该被被标记为None,这条线就停止计算
  4. # 如果到某一个节点,刚好是目标节点,则返回目标节点(不为空),向上返回,标记这条线为有效的
  5. if root is None or root == p or root == q:
  6. return root
  7. # 上面一步是判断到达某一个节点是是否要停止该条路径向下的迭代,如果不停止的话,应该做的事就是,通过判断当前节点的两条子路径
  8. # 的情况,也就是子路径里是否包含目标节点,来判断当前节点向上返回的状态,下面的left和right分别为左边和右边返回的子树
  9. # 的状态,分别标记是否有效
  10. left = self.lowestCommonAncestor(root.left, p, q)
  11. right = self.lowestCommonAncestor(root.right, p, q)
  12. # 如果这个时候左右子路径刚好包含所有的目标节点,那么就说明这个root节点就是要找的公共子节点,直接返回这个节点即可
  13. if left and right:
  14. return root
  15. # 如果这个节点不是公共祖先,那么它就从两颗子树那继承状态,并返回给上一级
  16. if left:
  17. return left
  18. else:
  19. return right

2、官方还给出了一种解题思路,就是用哈希表实现,下面就是用hashmap实现的思路。

        用存储父节点的方式实现两个节点的最近公共祖先的查找。这个方法很直观,首先通过遍历的方式存储每个节点的父节点。 然后通过遍历的方式获取其中一个节点的所有父节点。遍历另一个节点的所有父节点,直到找到两者的交集点,即为所求的公共祖先。

这个方法相对于上面那个直接迭代判断的方法,消耗的内存资源会多一点。

  1. class Solution:
  2. def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> bool:
  3. def dfs(root, parent):
  4. if root.left:
  5. parent[root.left.val] = root
  6. dfs(root.left, parent)
  7. if root.right:
  8. parent[root.right.val] = root
  9. dfs(root.right, parent)
  10. # 通过递归的方式,获取节点间父子关系的map
  11. parent = {}
  12. dfs(root, parent)
  13. parent[root.val] = None
  14. # 通过遍历的方式,找出其中一个目标节点到根节点的路劲
  15. visit_p = {}
  16. while p:
  17. visit_p[p] = True
  18. p = parent[p.val]
  19. while q:
  20. if visit_p.get(q, False):
  21. return q
  22. q = parent[q.val]

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/209779?site
推荐阅读
相关标签
  

闽ICP备14008679号