码迷,mamicode.com
首页 > 其他好文 > 详细

68 - II. 二叉树的最近公共祖先

时间:2020-03-22 19:44:21      阅读:67      评论:0      收藏:0      [点我收藏+]

标签:证明   sel   back   val   返回   方法   16px   problems   思考   

题目:给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。https://leetcode-cn.com/problems/er-cha-shu-de-zui-jin-gong-gong-zu-xian-lcof/submissions/

 写在前面:这一题不同于68-I 二叉搜索树的最近公共祖先https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-zui-jin-gong-gong-zu-xian-lcof/(二叉搜索树具有左节点<父节点<右节点的性质),那一题的解决思路以及代码如下:

# 解题思路 如果要查找的两个节点1.任一个等于根结点值,或者这两个节点指分别位于根节点值的两侧,则返回根节点。2.如果这两个节点只在根节点值的左侧,在左子树查找,3.否则在右子树查找。
# 注意返回的是节点而非节点的值
class Solution:
    def lowestCommonAncestor(self, root: TreeNode, p: TreeNode, q: TreeNode) -> TreeNode:
        if p.val>q.val:
            p,q = q,p
        if p==root or q == root:
            return root
        if p.val<root.val and q.val> root.val:
            return root
        if p.val<root.val and q.val<root.val:
            return self.lowestCommonAncestor(root.left,p,q)
        if p.val>root.val and q.val>root.val:
            return self.lowestCommonAncestor(root.right,p,q)

针对本题:

方法一.保存路径形式的思考方式:

第一种:假设给我们的树中的节点包含一个指向父节点的指针

如果有了父节点的指针,到达4的逆序路径能得出来为4->2->5->3,到达6的逆序路径为6->5->3由于都有父节点指针,所以这个问题就转换成了求两个链表的第一个公共节点问题;

第二种:使用数组存储路径

我们可以使用两个数组保存路径,用先序遍历存储正向路径,如节点4的正向路径为3->5->2->4,节点6的正向路径为3->5->6,从头找到第一个不相同节点的前一个节点即可,这个例子就是指的节点5;

第二种思路的代码:

# 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 lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        stack_1 = []
        stack_2 = []
        def dfs(root, node, stack):
            if not root:
                return False
            stack.append(root)
            if root.val == node.val:
                return True          
            if (dfs(root.left, node, stack) or dfs(root.right, node, stack)):
                return True
            stack.pop()
            
        dfs(root, p, stack_1)
        dfs(root, q, stack_2)
        i = 0
        while i < len(stack_1) and i<len(stack_2) and stack_1[i] == stack_2[i]:
            result = stack_1[i]
            i += 1
        return result

 

方法二.递归

【思路】

因为lowestCommonAncestor(root, p, q)的功能是找出以root为根节点的两个节点p和q的最近公共祖先,所以递归体分三种情况讨论:

1.如果p和q分别是root的左右节点,那么root就是我们要找的最近公共祖先;

2.如果p和q都是root的左节点,那么返回lowestCommonAncestor(root.left,p,q);

3.如果p和q都是root的右节点,那么返回lowestCommonAncestor(root.right,p,q);

边界条件讨论:

1.如果root是null,则说明我们已经找到最底了,返回null表示没找到;

2.如果root与p相等或者与q相等,则返回root;

3.如果左子树没找到,递归函数返回null,证明p和q同在root的右侧,那么最终的公共祖先就是右子树找到的结点;

4.如果右子树没找到,递归函数返回null,证明p和q同在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 lowestCommonAncestor(self, root, p, q):
        """
        :type root: TreeNode
        :type p: TreeNode
        :type q: TreeNode
        :rtype: TreeNode
        """
        if not root or root == p or root == q:
            return root
        left = self.lowestCommonAncestor(root.left, p, q)
        right = self.lowestCommonAncestor(root.right, p, q)

        if not left:
            return right
        if not right:
            return left
        return root

 

 

 

技术图片

 

68 - II. 二叉树的最近公共祖先

标签:证明   sel   back   val   返回   方法   16px   problems   思考   

原文地址:https://www.cnblogs.com/USTC-ZCC/p/12547620.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!