【LeetCode】236.二叉樹的最近公共祖先 (Python)

原文鏈接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/solution/shi-yong-zi-dian-cun-chu-shuang-qin-jie-dian-by-mi/

https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/

給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。

百度百科中最近公共祖先的定義爲:“對於有根樹 T 的兩個結點 p、q,最近公共祖先表示爲一個結點 x,滿足 x 是 p、q 的祖先且 x 的深度儘可能大(一個節點也可以是它自己的祖先)。”

例如,給定如下二叉樹:  root = [3,5,1,6,2,0,8,null,null,7,4]

 

示例 1:

輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
輸出: 3
解釋: 節點 5 和節點 1 的最近公共祖先是節點 3。
示例 2:

輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4
輸出: 5
解釋: 節點 5 和節點 4 的最近公共祖先是節點 5。因爲根據定義最近公共祖先節點可以爲節點本身。
 

說明:

所有節點的值都是唯一的。
p、q 爲不同節點且均存在於給定的二叉樹中。

# 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
        """
        m = {}
        def dfs(root):
            if root:
                if root.left:
                    m[root.left] = root
                if root.right:
                    m[root.right] = root
                dfs(root.left)
                dfs(root.right)
                
        dfs(root)
        l1, l2 = p, q
        while l1 != l2:
            l1 = m.get(l1, None) if m.get(l1, None) is not None else q
            l2 = m.get(l2, None) if m.get(l2, None) is not None else p
            
        return l1
        

思路:

由於每個節點只有唯一一個父節點,我們可以使用到字典的value-key的形式(節點-父節點)字典中預置根節點的父節點爲None。字典建立完成後,二叉樹就可以看成一個所有節點都將最終指向根節點的鏈表了。於是在二叉樹中尋找兩個節點的最小公共節點就相當於,在一個鏈表中尋找他們相遇的節點。

作者:mian-mian-sir
鏈接:https://leetcode-cn.com/problems/lowest-common-ancestor-of-a-binary-tree/solution/shi-yong-zi-dian-cun-chu-shuang-qin-jie-dian-by-mi/

上面代碼中dfs爲深度遍歷二叉樹,建立子節點指向根節點的字典。其中根節點root我沒建立,後面可以默認None,然後重點來了,將兩個子節點分別同時尋找它的根節點,如果兩個節點能夠相等,說明已經找到根節點,如果一個節點發現沒有父節點,說明是上面圖示的情況,即一個節點離LCA近,一個離LCA遠。那麼我們開始遍歷每個節點的根節點,當l2已經到達根節點root時,l1還沒有到,並且此刻l1和l2的距離和一開始l1,l2距離是一樣的,那麼此時將l1初始的位置賦值給l2,繼續遍歷;當l1到達根節點時,l2達到到達與初始l2相同的距離(注意不是相同的位置,是距離根節點一樣的距離),此時將l2的初始位置賦值給l1,那麼此時l1和l2距離LCA就是一樣的距離了,那麼現在只要兩個節點繼續遍歷下去就肯定能找到LCA。

此時將另一個節點賦值給該節點,然後繼續尋找父節點;之後另一個節點也會發現沒有父節點,則將

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章